diff --git a/setup.py b/setup.py
index a4be5f9..2527a4f 100644
--- a/setup.py
+++ b/setup.py
@@ -13,7 +13,7 @@ with open('requirements.txt') as f:
 
 setup(
     name='django-clickhouse',
-    version='1.0.0',
+    version='1.0.1',
     packages=['django_clickhouse'],
     package_dir={'': 'src'},
     url='https://github.com/carrotquest/django-clickhouse',
diff --git a/src/django_clickhouse/migrations.py b/src/django_clickhouse/migrations.py
index 62ccd18..9d47688 100644
--- a/src/django_clickhouse/migrations.py
+++ b/src/django_clickhouse/migrations.py
@@ -7,8 +7,13 @@ from typing import Optional, Set
 from django.db import DEFAULT_DB_ALIAS as DJANGO_DEFAULT_DB_ALIAS
 from django.db.models.signals import post_migrate
 from django.dispatch import receiver
-from infi.clickhouse_orm.database import ServerError, DatabaseException
+
+# In order to support all operations import here
 from infi.clickhouse_orm.migrations import *
+
+from infi.clickhouse_orm.database import ServerError, DatabaseException
+from infi.clickhouse_orm.fields import StringField, DateField
+from infi.clickhouse_orm.engines import MergeTree
 from infi.clickhouse_orm.utils import import_submodules
 
 from .clickhouse_models import ClickHouseModel