diff --git a/SAS/TMSS/backend/test/t_scheduling_units.py b/SAS/TMSS/backend/test/t_scheduling_units.py
index f3b00444cadda36061f1057ab0361db5a72a2696..1f998c43e6a40205e9c22a293ef0fc7d5f2b0e73 100644
--- a/SAS/TMSS/backend/test/t_scheduling_units.py
+++ b/SAS/TMSS/backend/test/t_scheduling_units.py
@@ -1317,7 +1317,7 @@ class SchedulingUnitBlueprintIndirectModificationsTestCase(unittest.TestCase):
             client.create_cleanuptask_for_scheduling_unit_blueprint(scheduling_unit_blueprint['id'])
 
             # check the blueprint graph
-            scheduling_unit_blueprint = client.get_schedulingunit_blueprint(scheduling_unit_blueprint['id'])
+            scheduling_unit_blueprint = client.get_schedulingunit_blueprint(scheduling_unit_blueprint['id'], extended=True)
             self.assertEqual(num_task_blueprints_before + 1, len(scheduling_unit_blueprint['task_blueprints']))
             self.assertIn("Cleanup", [task['name'] for task in scheduling_unit_blueprint['task_blueprints']])
             for task in scheduling_unit_blueprint['task_blueprints']:
@@ -1336,7 +1336,7 @@ class SchedulingUnitBlueprintIndirectModificationsTestCase(unittest.TestCase):
 
             # create another cleanup task for the scheduling unit and check name is as expected
             client.create_cleanuptask_for_scheduling_unit_blueprint(scheduling_unit_blueprint['id'])
-            scheduling_unit_blueprint = client.get_schedulingunit_blueprint(scheduling_unit_blueprint['id'])
+            scheduling_unit_blueprint = client.get_schedulingunit_blueprint(scheduling_unit_blueprint['id'], extended=True)
             self.assertIn("Cleanup2", [task['name'] for task in scheduling_unit_blueprint['task_blueprints']])
 
     def test_TMSS_1585_bugfix_check_cleanup_starts(self):