diff --git a/test/unit/jobs/test_job_wrapper.py b/test/unit/jobs/test_job_wrapper.py
index b7f3560c89a..c5e3b5e6e20 100644
--- a/test/unit/jobs/test_job_wrapper.py
+++ b/test/unit/jobs/test_job_wrapper.py
@@ -31,7 +31,7 @@ class BaseWrapperTestCase(UsesApp):
         job.user = User()
         job.object_store_id = "foo"
         self.model_objects = {Job: {345: job}}
-        self.app.model.context = MockContext(self.model_objects)
+        self.app.model._session = MockContext(self.model_objects)
 
         self.app.toolbox = MockToolbox(MockTool(self))
         self.working_directory = os.path.join(self.test_directory, "working")