diff --git a/tests/test_instance_service.py b/tests/test_instance_service.py
index 5011b0f..70cac98 100644
--- a/tests/test_instance_service.py
+++ b/tests/test_instance_service.py
@@ -31,6 +31,9 @@ from govoplan_workflow.backend.db.models import (
     WorkflowInstance,
     WorkflowInstanceEvent,
     WorkflowInstanceStep,
+    WorkflowTrigger,
+    WorkflowTriggerDelivery,
+    WorkflowWaitState,
 )
 from govoplan_workflow.backend.instance_service import (
     SqlWorkflowRuntimeWorker,
@@ -387,6 +390,9 @@ class WorkflowInstanceServiceTests(unittest.TestCase):
                 WorkflowInstance.__table__,
                 WorkflowInstanceStep.__table__,
                 WorkflowInstanceEvent.__table__,
+                WorkflowTrigger.__table__,
+                WorkflowTriggerDelivery.__table__,
+                WorkflowWaitState.__table__,
             ],
         )
         self.Session = sessionmaker(bind=self.engine)
@@ -421,7 +427,10 @@ class WorkflowInstanceServiceTests(unittest.TestCase):
             tables=[
                 WorkflowInstanceEvent.__table__,
                 WorkflowInstanceStep.__table__,
+                WorkflowWaitState.__table__,
                 WorkflowInstance.__table__,
+                WorkflowTriggerDelivery.__table__,
+                WorkflowTrigger.__table__,
                 WorkflowDefinitionRevision.__table__,
                 WorkflowDefinition.__table__,
             ],