diff --git a/superxas_bec/devices/timepix/timepix.py b/superxas_bec/devices/timepix/timepix.py
index e82f8a8..ef38649 100644
--- a/superxas_bec/devices/timepix/timepix.py
+++ b/superxas_bec/devices/timepix/timepix.py
@@ -17,7 +17,7 @@ import numpy as np
 from bec_lib.logger import bec_logger
 from ophyd import ADBase
 from ophyd import Component as Cpt
-from ophyd import DeviceStatus, Kind, StatusBase
+from ophyd import Device, DeviceStatus, Kind, StatusBase
 from ophyd_devices import AndStatus as _AndStatus
 from ophyd_devices import AsyncSignal, CompareStatus, TransitionStatus
 from ophyd_devices.devices.areadetector.cam import ASItpxCam
@@ -31,9 +31,6 @@ from superxas_bec.devices.timepix.timepix_fly_client.timepix_fly_interface impor
     PixelMap,
 )
 
-if TYPE_CHECKING:  # pragma: no cover
-    from ophyd import Device
-
 
 class AndStatus(StatusBase):
     """Custom AndStatus for TimePix detector."""