diff --git a/categories/management/commands/__init__.py b/categories/management/commands/__init__.py
index 5da0f94..0721d7c 100644
--- a/categories/management/commands/__init__.py
+++ b/categories/management/commands/__init__.py
@@ -1,4 +1,8 @@
-from django.db.models.signals import post_syncdb
-from categories.migration import migrate_app
+try:
+    from south.db import db
+    from django.db.models.signals import post_syncdb
+    from categories.migration import migrate_app
 
-post_syncdb.connect(migrate_app)
+    post_syncdb.connect(migrate_app)
+except ImportError:
+    pass
diff --git a/categories/registration.py b/categories/registration.py
index e0620f2..05d865b 100644
--- a/categories/registration.py
+++ b/categories/registration.py
@@ -3,7 +3,6 @@ These functions handle the adding of fields to other models
 """
 from django.db.models import FieldDoesNotExist
 import fields
-
 from settings import FIELD_REGISTRY, MODEL_REGISTRY
 
 
diff --git a/categories/tests/__init__.py b/categories/tests/__init__.py
index b144d87..194e08d 100644
--- a/categories/tests/__init__.py
+++ b/categories/tests/__init__.py
@@ -1,5 +1,6 @@
 from categories.tests.category_import import *
 from categories.tests.templatetags import *
 from categories.tests.manager import *
+from categories.tests.registration import *
 
 __fixtures__ = ['categories.json']