diff --git a/tests/clickhouse_migrations/0001_initial.py b/tests/clickhouse_migrations/0001_initial.py
index d9c0277..d7bf69a 100644
--- a/tests/clickhouse_migrations/0001_initial.py
+++ b/tests/clickhouse_migrations/0001_initial.py
@@ -1,10 +1,10 @@
 
-from django_clickhouse import migrations
+from django_clickhouse import migrations, migration_operators
 from tests.clickhouse_models import ClickHouseTestModel, ClickHouseCollapseTestModel
 
 
 class Migration(migrations.Migration):
     operations = [
-        migrations.CreateTable(ClickHouseTestModel),
-        migrations.CreateTable(ClickHouseCollapseTestModel)
+        migration_operators.CreateTable(ClickHouseTestModel),
+        migration_operators.CreateTable(ClickHouseCollapseTestModel)
     ]