diff --git a/backend/apps/archive/apps.py b/backend/apps/archive/apps.py
index fa0ab90..02fab5d 100644
--- a/backend/apps/archive/apps.py
+++ b/backend/apps/archive/apps.py
@@ -11,7 +11,7 @@ class ArchiveConfig(AppConfig):
 
 @receiver(post_migrate)
 def insert_defualt_data(sender, **kwargs):
-    from apps.archive.models import CategoryModel
+    from apps.archive.models import SourceSiteModel
 
     categories = {
         "furaffinity": "FurAffinity",
@@ -21,4 +21,4 @@ def insert_defualt_data(sender, **kwargs):
 
     if sender.name == ArchiveConfig.name:
         for key, value in categories.items():
-            CategoryModel.objects.get_or_create(slug=key, name=value)
+            SourceSiteModel.objects.get_or_create(slug=key, name=value)