diff --git a/pioreactor/automations/dosing/base.py b/pioreactor/automations/dosing/base.py
index 08e7af70..36780b9c 100644
--- a/pioreactor/automations/dosing/base.py
+++ b/pioreactor/automations/dosing/base.py
@@ -19,9 +19,9 @@
 from pioreactor.actions.pump import add_alt_media
 from pioreactor.actions.pump import add_media
 from pioreactor.actions.pump import remove_waste
+from pioreactor.automations import BaseAutomationJob
 from pioreactor.automations import events
 from pioreactor.background_jobs.dosing_control import DosingController
-from pioreactor.background_jobs.subjobs import BackgroundSubJob
 from pioreactor.config import config
 from pioreactor.pubsub import QOS
 from pioreactor.utils import is_pio_job_running
@@ -157,7 +157,7 @@ def _update_alt_media_fraction(
         )
 
 
-class DosingAutomationJob(BackgroundSubJob):
+class DosingAutomationJob(BaseAutomationJob):
     """
     This is the super class that automations inherit from. The `run` function will
     execute every `duration` minutes (selected at the start of the program). If `duration` is left
diff --git a/pioreactor/automations/temperature/base.py b/pioreactor/automations/temperature/base.py
index 925c11f8..b792e28d 100644
--- a/pioreactor/automations/temperature/base.py
+++ b/pioreactor/automations/temperature/base.py
@@ -11,14 +11,14 @@
 from pioreactor import exc
 from pioreactor import structs
 from pioreactor import types as pt
-from pioreactor.background_jobs.subjobs import BackgroundSubJob
+from pioreactor.automations import BaseAutomationJob
 from pioreactor.background_jobs.temperature_control import TemperatureController
 from pioreactor.pubsub import QOS
 from pioreactor.utils import is_pio_job_running
 from pioreactor.utils.timing import current_utc_datetime
 
 
-class TemperatureAutomationJob(BackgroundSubJob):
+class TemperatureAutomationJob(BaseAutomationJob):
     """
     This is the super class that Temperature automations inherit from.
     The `execute` function, which is what subclasses will define, is updated every time a new temperature is recorded to MQTT.