diff --git a/tests/python/eevee_next_render_tests.py b/tests/python/eevee_next_render_tests.py
index 2348357bd59..ed9a5e37f1b 100644
--- a/tests/python/eevee_next_render_tests.py
+++ b/tests/python/eevee_next_render_tests.py
@@ -52,6 +52,13 @@ def setup():
         # Light-probes
         eevee.gi_cubemap_resolution = '256'
 
+        # Only include the plane in probes
+        for ob in scene.objects:
+            if ob.name != 'Plane' and ob.type != 'LIGHT':
+                ob.hide_probe_volume = True
+                ob.hide_probe_sphere = True
+                ob.hide_probe_plane = True
+
         # Does not work in edit mode
         if bpy.context.mode == 'OBJECT':
             # Simple probe setup
@@ -73,13 +80,6 @@ def setup():
             grid.data.grid_dilation_threshold = 1.0
             bpy.ops.object.lightprobe_cache_bake(subset='ACTIVE')
 
-        # Only include the plane in probes
-        for ob in scene.objects:
-            if ob.name != 'Plane' and ob.type != 'LIGHT':
-                ob.hide_probe_volume = True
-                ob.hide_probe_sphere = True
-                ob.hide_probe_plane = True
-
 
 # When run from inside Blender, render and exit.
 try: