diff --git a/SAS/TMSS/backend/test/t_scheduling.py b/SAS/TMSS/backend/test/t_scheduling.py
index 2c7a7c33c75c3029b695390bbf1d15452dd1089f..577c82bdceebd8eef12cdd9e64e5b468e60dc58b 100755
--- a/SAS/TMSS/backend/test/t_scheduling.py
+++ b/SAS/TMSS/backend/test/t_scheduling.py
@@ -75,6 +75,7 @@ from lofar.sas.resourceassignment.resourceassigner.rarpc import RARPC
 from lofar.sas.tmss.tmss.tmssapp import models
 from lofar.sas.tmss.tmss.tmssapp.subtasks import *
 from lofar.sas.tmss.tmss.tmssapp.tasks import *
+from lofar.sas.tmss.test.test_utils import set_subtask_state_following_allowed_transitions
 from lofar.messaging.rpc import RPCService, ServiceMessageHandler
 import threading