diff --git a/awx/main/models/__init__.py b/awx/main/models/__init__.py
index b1912cdf14..848db3b593 100644
--- a/awx/main/models/__init__.py
+++ b/awx/main/models/__init__.py
@@ -7,7 +7,7 @@ from django.db.models.signals import pre_delete  # noqa
 
 # AWX
 from awx.main.models.base import (  # noqa
-    BaseModel, prevent_search, CLOUD_INVENTORY_SOURCES, VERBOSITY_CHOICES
+    BaseModel, PrimordialModel, prevent_search, CLOUD_INVENTORY_SOURCES, VERBOSITY_CHOICES
 )
 from awx.main.models.unified_jobs import (  # noqa
     UnifiedJob, UnifiedJobTemplate, StdoutMaxBytesExceeded
diff --git a/tools/data_generators/rbac_dummy_data_generator.py b/tools/data_generators/rbac_dummy_data_generator.py
index 1e8ecf0bcf..4b92a4c0e2 100755
--- a/tools/data_generators/rbac_dummy_data_generator.py
+++ b/tools/data_generators/rbac_dummy_data_generator.py
@@ -40,13 +40,11 @@ from django.db import transaction # noqa
 # awx
 from awx.main.models import (  # noqa
     Credential, CredentialType, Group, Host, Inventory, Job, JobEvent,
-    JobHostSummary, JobTemplate, Label, Organization, Project,
+    JobHostSummary, JobTemplate, Label, Organization, PrimordialModel, Project,
     Team, User, WorkflowJobTemplate, WorkflowJobTemplateNode,
     batch_role_ancestor_rebuilding,
 )
 
-from awx.main.models.base import PrimordialModel
-
 from awx.main.signals import ( # noqa
     disable_activity_stream,
     disable_computed_fields