diff --git a/awx/main/tests/unit/scheduler/conftest.py b/awx/main/tests/unit/scheduler/conftest.py
index 6b07649fd0..97eec0dbb4 100644
--- a/awx/main/tests/unit/scheduler/conftest.py
+++ b/awx/main/tests/unit/scheduler/conftest.py
@@ -22,9 +22,10 @@ def epoch():
 
 @pytest.fixture
 def scheduler_factory(mocker, epoch):
+    mocker.patch('awx.main.models.Instance.objects.total_capacity', return_value=999999999)
+
     def fn(tasks=[], inventory_sources=[], latest_project_updates=[], latest_inventory_updates=[], create_project_update=None, create_inventory_update=None):
         sched = TaskManager()
-        sched.capacity_total = 999999999
 
         sched.graph.get_now = lambda: epoch