1 files changed, 4 insertions, 2 deletions
diff --git a/ishtar_common/tests.py b/ishtar_common/tests.py
index 122f29d92..6d112f2d1 100644
--- a/ishtar_common/tests.py
+++ b/ishtar_common/tests.py
@@ -682,6 +682,8 @@ class GenericSerializationTest:
                 module = importlib.import_module(module_name + ".models")
 
             model = getattr(module, model_name)
+            if getattr(model, "TO_BE_DELETED", False):
+                continue
             current_count = model.objects.count()
             result = json.loads(json_result[key])
             serialization_count = len(result)
@@ -917,11 +919,11 @@ class SerializationTest(GenericSerializationTest, TestCase):
         )
         wl1 = WarehouseDivisionLink.objects.create(
             warehouse=w1,
-            division=wd1
+            container_type=ContainerType.objects.all()[0],
         )
         wl2 = WarehouseDivisionLink.objects.create(
             warehouse=w2,
-            division=wd2
+            container_type = ContainerType.objects.all()[1],
         )
         ContainerLocalisation.objects.create(
             container=c1,