diff --git a/tests/utils_tests/models.py b/tests/utils_tests/models.py
index 866a37debc44..ff8ad1031c7d 100644
--- a/tests/utils_tests/models.py
+++ b/tests/utils_tests/models.py
@@ -1,4 +1,6 @@
 from django.db import models
+from django_singlestore.schema import ModelStorageManager
+
 
 
 class Category(models.Model):
@@ -7,3 +9,4 @@ class Category(models.Model):
 
 class CategoryInfo(models.Model):
     category = models.OneToOneField(Category, models.CASCADE)
+    objects = ModelStorageManager("REFERENCE")