diff --git a/src/dodal/devices/pressure_jump_cell.py b/src/dodal/devices/pressure_jump_cell.py
index 70944b37fb..d32ea0019d 100644
--- a/src/dodal/devices/pressure_jump_cell.py
+++ b/src/dodal/devices/pressure_jump_cell.py
@@ -216,10 +216,14 @@ def __init__(
         adc_prefix: str = "",
         name: str = "",
     ):
-        self.all_valves_control = AllValvesControl(f"{beamline_prefix}{cell_prefix}", name)
+        self.all_valves_control = AllValvesControl(
+            f"{beamline_prefix}{cell_prefix}", name
+        )
         self.pump = Pump(f"{beamline_prefix}{cell_prefix}", name)
 
-        self.controller = PressureJumpCellController(f"{beamline_prefix}{cell_prefix}", name)
+        self.controller = PressureJumpCellController(
+            f"{beamline_prefix}{cell_prefix}", name
+        )
 
         with self.add_children_as_readables():
             self.pressure_transducers: DeviceVector[PressureTransducer] = DeviceVector(
@@ -233,6 +237,8 @@ def __init__(
                 }
             )
 
-            self.cell_temperature = epics_signal_r(float, f"{beamline_prefix}{cell_prefix}TEMP")
+            self.cell_temperature = epics_signal_r(
+                float, f"{beamline_prefix}{cell_prefix}TEMP"
+            )
 
         super().__init__(name)
diff --git a/tests/devices/unit_tests/test_pressure_jump_cell.py b/tests/devices/unit_tests/test_pressure_jump_cell.py
index 9f5128c709..8f2a611f55 100644
--- a/tests/devices/unit_tests/test_pressure_jump_cell.py
+++ b/tests/devices/unit_tests/test_pressure_jump_cell.py
@@ -4,12 +4,9 @@
 from ophyd_async.core import DeviceCollector, assert_reading, set_mock_value
 
 from dodal.devices.pressure_jump_cell import (
-    BusyState,
     FastValveState,
-    LimitSwitchState,
     PressureJumpCell,
     PumpMotorDirectionState,
-    TimerState,
     ValveState,
 )