diff --git a/tests/unit_tests/device_setup_plans/test_setup_oav.py b/tests/unit_tests/device_setup_plans/test_setup_oav.py
index 8cdb72b59..af77078b0 100644
--- a/tests/unit_tests/device_setup_plans/test_setup_oav.py
+++ b/tests/unit_tests/device_setup_plans/test_setup_oav.py
@@ -93,7 +93,7 @@ def test_when_set_up_oav_with_different_zoom_levels_then_flat_field_applied_corr
     RE = RunEngine()
     RE(pre_centring_setup_oav(oav, mock_parameters, ophyd_pin_tip_detection))
     assert oav.mxsc.input_plugin.get() == expected_plugin
-    assert oav.snapshot.input_plugin.get() == "OAV.MXSC"
+    assert oav.snapshot.input_plugin.get() == expected_plugin
 
 
 @pytest.mark.parametrize(