diff --git a/tests/models.py b/tests/models.py
index 78c8e21..a0de1ec 100644
--- a/tests/models.py
+++ b/tests/models.py
@@ -3,7 +3,7 @@ This file contains sample models to use in tests
 """
 from django.db import models
 from django.db.models.manager import BaseManager
-from django_pg_returning.models import UpdateReturningModel
+from django_pg_returning import UpdateReturningModel
 
 from django_clickhouse.models import ClickHouseSyncModel, ClickHouseSyncQuerySet