diff --git a/test/integration/test_figure_scripts.py b/test/integration/test_figure_scripts.py
index bdd55085a..c5e3a4288 100644
--- a/test/integration/test_figure_scripts.py
+++ b/test/integration/test_figure_scripts.py
@@ -59,15 +59,10 @@ def test_export_figure_as(self, export_option, tmpdir):
         image = self.create_test_image(size_x, size_y, size_z, size_c,
                                        size_t, session)
 
-        image_id = image.getId()  # self.import_pyramid(tmpdir, client=client)
-        query_service = client.sf.getQueryService()
-        big_image = query_service.findByQuery(
-            """select i from Image i left outer join fetch i.pixels as p
-               where i.id = :id""",
-            ParametersI().addId(image_id))
+        image_id = image.getId()
 
         figure_name = "test_export_figure_as_%s" % export_option
-        json = create_figure([image, big_image])
+        json = create_figure([image])
         uri = "https://www.openmicroscopy.org/"
         args = {
             "Figure_JSON": omero.rtypes.rstring(json),