diff --git a/netbox/dcim/migrations/0191_module_bay_rebuild.py b/netbox/dcim/migrations/0191_module_bay_rebuild.py
index 86a57c300..7944c31aa 100644
--- a/netbox/dcim/migrations/0191_module_bay_rebuild.py
+++ b/netbox/dcim/migrations/0191_module_bay_rebuild.py
@@ -2,7 +2,7 @@ from django.db import migrations
 
 
 def rebuild_mptt(apps, schema_editor):
-    ModuleBay = apps.get_model('dcim', 'ModuleBay')
+    from dcim.models import ModuleBay
     ModuleBay.objects.rebuild()