diff --git a/tests/test_scheduled_job.py b/tests/test_scheduled_job.py
index 49dbdd7279..ad2bf735ab 100644
--- a/tests/test_scheduled_job.py
+++ b/tests/test_scheduled_job.py
@@ -9,7 +9,6 @@
 from portal.models.role import ROLE
 from portal.models.scheduled_job import ScheduledJob
 from portal.tasks import test as test_task
-from sqlalchemy.orm import session
 from tests import TestCase
 
 
@@ -176,4 +175,4 @@ def test_job_trigger(self):
         resp = self.client.post('/api/scheduled_job/{}/trigger'.format(job.id))
         assert resp.status_code == 200
 
-        session.close_all_sessions()
+        db.session.close_all()