diff --git a/test/database_service/test_db_experiment_data.py b/test/database_service/test_db_experiment_data.py
index 3cbfa52928..3d83fa8512 100644
--- a/test/database_service/test_db_experiment_data.py
+++ b/test/database_service/test_db_experiment_data.py
@@ -409,6 +409,8 @@ def test_get_figure(self):
         exp_data = ExperimentData(experiment_type="qiskit_test")
         figure_template = "hello world {}"
         name_template = "figure_{}.svg"
+        name_template_wo_ext = "figure_{}"
+
         for idx in range(3):
             exp_data.add_figures(
                 str.encode(figure_template.format(idx)), figure_names=name_template.format(idx)
@@ -418,6 +420,11 @@ def test_get_figure(self):
         self.assertEqual(expected_figure, exp_data.figure(name_template.format(idx)).figure)
         self.assertEqual(expected_figure, exp_data.figure(idx).figure)
 
+        # Check that figure will be returned without file extension in name
+        expected_figure = str.encode(figure_template.format(idx))
+        self.assertEqual(expected_figure, exp_data.figure(name_template_wo_ext.format(idx)).figure)
+        self.assertEqual(expected_figure, exp_data.figure(idx).figure)
+
         file_name = uuid.uuid4().hex
         self.addCleanup(os.remove, file_name)
         exp_data.figure(idx, file_name)