diff --git a/opennebula/models.py b/opennebula/models.py
index 2730474..6c7dc52 100644
--- a/opennebula/models.py
+++ b/opennebula/models.py
@@ -1,7 +1,6 @@
 import uuid
 from django.db import models
 from django.contrib.auth import get_user_model
-from django.contrib.postgres.fields import JSONField
 from uncloud_pay.models import Product
 
 # ungleich specific
@@ -13,7 +12,7 @@ storage_class_mapping = {
 
 class VM(Product):
     vmid = models.IntegerField(primary_key=True)
-    data = JSONField()
+    data = models.JSONField()
 
     @property
     def uncloud_name(self):
diff --git a/uncloud/models.py b/uncloud/models.py
index bd7a931..212d555 100644
--- a/uncloud/models.py
+++ b/uncloud/models.py
@@ -1,5 +1,6 @@
 from django.db import models
-from django.contrib.postgres.fields import JSONField
+from django.db.models import JSONField
+
 from django.utils.translation import gettext_lazy as _
 
 class UncloudModel(models.Model):