diff --git a/SAS/TMSS/backend/test/t_subtask_validation.py b/SAS/TMSS/backend/test/t_subtask_validation.py
index 2abd4418e535a5aeb6c8bbd1e91bcd7d49acb876..7d7dc1d1fb5410ba8e24f4efb812eb2d8b9fb059 100755
--- a/SAS/TMSS/backend/test/t_subtask_validation.py
+++ b/SAS/TMSS/backend/test/t_subtask_validation.py
@@ -68,11 +68,11 @@ class SubtaskValidationTest(unittest.TestCase):
         subtask_template = self.create_subtask_template(minimal_json_schema())
         specifications_doc = '{ this is not a json object }'
         subtask_data = Subtask_test_data(subtask_template=subtask_template, specifications_doc=specifications_doc,
-                                         cluster=self.cluster, state=self.state)
+                                         cluster=self.cluster, state=self.state,
+                                         task_blueprint=self.task_blueprint)
 
         with self.assertRaises(SchemaValidationException) as context:
             subtask = models.Subtask.objects.create(**subtask_data)
-            subtask.task_blueprints.set([self.task_blueprint])
         self.assertTrue('invalid json' in str(context.exception).lower())
 
     def test_validate_correlator_schema_with_valid_specification(self):
diff --git a/SAS/TMSS/backend/test/t_subtasks.py b/SAS/TMSS/backend/test/t_subtasks.py
index 676dac6223bb58372aca51f07937fe187b3c1a3d..3606c045aa60aca327caa9835635831becc20a40 100755
--- a/SAS/TMSS/backend/test/t_subtasks.py
+++ b/SAS/TMSS/backend/test/t_subtasks.py
@@ -50,9 +50,8 @@ def create_subtask_object_for_testing(subtask_type_value):
     """
     template_type = models.SubtaskType.objects.get(value=subtask_type_value)
     subtask_template_obj = create_subtask_template_for_testing(template_type)
-    subtask_data = Subtask_test_data(subtask_template=subtask_template_obj)
+    subtask_data = Subtask_test_data(subtask_template=subtask_template_obj, task_blueprint=models.TaskBlueprint.objects.create(**TaskBlueprint_test_data()))
     subtask = models.Subtask.objects.create(**subtask_data)
-    subtask.task_blueprints.set([models.TaskBlueprint.objects.create(**TaskBlueprint_test_data())])
     return subtask
 
 def create_subtask_template_for_testing(template_type: object):