diff --git a/ibu/backends/mysql/base.py b/ibu/backends/mysql/base.py
index 85a1d28..4dd460a 100755
--- a/ibu/backends/mysql/base.py
+++ b/ibu/backends/mysql/base.py
@@ -332,8 +332,7 @@ def check_constraints(self, table_names=None):
             table_names = self.introspection.table_names(cursor)
         for table_name in table_names:
             primary_key_column_name = self.introspection\
-                .get_primary_key_column(
-                                        cursor, table_name)
+                .get_primary_key_column(cursor, table_name)
             if not primary_key_column_name:
                 continue
             key_columns = self.introspection.get_key_columns(
diff --git a/ibu/backends/mysql/client.py b/ibu/backends/mysql/client.py
index 29059b0..467a984 100755
--- a/ibu/backends/mysql/client.py
+++ b/ibu/backends/mysql/client.py
@@ -1,6 +1,6 @@
 import subprocess
 
-from django.db.backends.base.client import BaseDatabaseClient
+from ibu.backends.base.client import BaseDatabaseClient
 
 
 class DatabaseClient(BaseDatabaseClient):