diff --git a/glue_jupyter/bqplot/image/tests/test_visual.py b/glue_jupyter/bqplot/image/tests/test_visual.py
index ce84b2d2..19fc02ee 100644
--- a/glue_jupyter/bqplot/image/tests/test_visual.py
+++ b/glue_jupyter/bqplot/image/tests/test_visual.py
@@ -22,11 +22,11 @@ def test_visual_incompatible_subset(
     app = jglue()
     data1 = app.add_data(image={"image": im})[0]
     data2 = app.add_data(catalog={"x": x, "y": y})[0]
+    image = app.imshow(data=data1, show=False)
     state1 = data1.pixel_component_ids[1] > 32
     app.data_collection.new_subset_group('image[x] > 32', state1)
     state2 = data2.id['x'] > 1
     app.data_collection.new_subset_group('x > 1', state2)
-    image = app.imshow(data=data1, show=False)
     figure = image.figure_widget
     figure.layout = {"width": "400px", "height": "250px"}
     return figure
diff --git a/glue_jupyter/tests/helpers.py b/glue_jupyter/tests/helpers.py
index 4d329b41..e0793d60 100644
--- a/glue_jupyter/tests/helpers.py
+++ b/glue_jupyter/tests/helpers.py
@@ -49,8 +49,8 @@ def test_wrapper(tmp_path, page_session, *args, **kwargs):
             viewer = page_session.locator(".test-viewer")
             viewer.wait_for()
 
-            # Wait for an additional 2 seconds
-            page_session.wait_for_timeout(2000)
+            # Wait for an additional 5 seconds
+            page_session.wait_for_timeout(5000)
 
             screenshot = viewer.screenshot()