diff --git a/alfalfa_worker/jobs/openstudio/create_run.py b/alfalfa_worker/jobs/openstudio/create_run.py
index f047f0b0..ff391801 100644
--- a/alfalfa_worker/jobs/openstudio/create_run.py
+++ b/alfalfa_worker/jobs/openstudio/create_run.py
@@ -4,7 +4,8 @@
 
 from alfalfa_worker.jobs.openstudio import lib_dir
 from alfalfa_worker.lib.enums import RunStatus, SimType
-from alfalfa_worker.lib.job import Job, JobExceptionInvalidModel
+from alfalfa_worker.lib.job import Job
+from alfalfa_worker.lib.job_exception import JobExceptionInvalidModel
 from alfalfa_worker.lib.utils import rel_symlink
 
 
diff --git a/alfalfa_worker/jobs/step_run_base.py b/alfalfa_worker/jobs/step_run_base.py
index cc2c981b..210732e8 100644
--- a/alfalfa_worker/jobs/step_run_base.py
+++ b/alfalfa_worker/jobs/step_run_base.py
@@ -7,11 +7,10 @@
 
 from alfalfa_worker.lib.constants import DATETIME_FORMAT
 from alfalfa_worker.lib.enums import AutoName, RunStatus
-from alfalfa_worker.lib.job import (
-    Job,
+from alfalfa_worker.lib.job import Job, message
+from alfalfa_worker.lib.job_exception import (
     JobException,
-    JobExceptionSimulation,
-    message
+    JobExceptionSimulation
 )
 from alfalfa_worker.lib.utils import to_bool