diff --git a/tests/tasks/uipath-maestro-case/sla_response/at_risk_enter_stage/check_at_risk_enter_stage.py b/tests/tasks/uipath-maestro-case/sla_response/at_risk_enter_stage/check_at_risk_enter_stage.py
index a5a36d0d4d..eb9db2d642 100644
--- a/tests/tasks/uipath-maestro-case/sla_response/at_risk_enter_stage/check_at_risk_enter_stage.py
+++ b/tests/tasks/uipath-maestro-case/sla_response/at_risk_enter_stage/check_at_risk_enter_stage.py
@@ -5,6 +5,7 @@
 import sys
 
 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
+from _shared.case_check import is_non_required  # noqa: E402
 from _shared.sla_response_check import (  # noqa: E402
     assert_at_risk_shape,
     assert_interrupting,
@@ -13,7 +14,6 @@
     assert_stage_count,
     fail,
     iter_sla_status_change,
-    is_non_required,
     label_of,
     read_plan,
     secondary_stages,
diff --git a/tests/tasks/uipath-maestro-case/sla_response/case_breach_oversight/check_case_breach_oversight.py b/tests/tasks/uipath-maestro-case/sla_response/case_breach_oversight/check_case_breach_oversight.py
index 0ecd0ddc8e..01b7cb998b 100644
--- a/tests/tasks/uipath-maestro-case/sla_response/case_breach_oversight/check_case_breach_oversight.py
+++ b/tests/tasks/uipath-maestro-case/sla_response/case_breach_oversight/check_case_breach_oversight.py
@@ -10,6 +10,7 @@
 import sys
 
 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
+from _shared.case_check import is_non_required  # noqa: E402
 from _shared.sla_response_check import (  # noqa: E402
     assert_breach_shape,
     assert_interrupting,
@@ -19,7 +20,6 @@
     fail,
     is_secondary,
     iter_sla_status_change,
-    is_non_required,
     label_of,
     read_plan,
 )
diff --git a/tests/tasks/uipath-maestro-case/sla_response/case_breach_takeover/check_case_breach_takeover.py b/tests/tasks/uipath-maestro-case/sla_response/case_breach_takeover/check_case_breach_takeover.py
index 6787afa7d0..ab6813e529 100644
--- a/tests/tasks/uipath-maestro-case/sla_response/case_breach_takeover/check_case_breach_takeover.py
+++ b/tests/tasks/uipath-maestro-case/sla_response/case_breach_takeover/check_case_breach_takeover.py
@@ -5,6 +5,7 @@
 import sys
 
 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
+from _shared.case_check import is_non_required  # noqa: E402
 from _shared.sla_response_check import (  # noqa: E402
     assert_breach_shape,
     assert_interrupting,
@@ -13,7 +14,6 @@
     assert_stage_count,
     fail,
     iter_sla_status_change,
-    is_non_required,
     label_of,
     read_plan,
     secondary_stages,