diff --git a/src/isar_robot/inspections.py b/src/isar_robot/inspections.py
index 022dbd0..fe491d5 100644
--- a/src/isar_robot/inspections.py
+++ b/src/isar_robot/inspections.py
@@ -7,13 +7,15 @@
 from robot_interface.models.exceptions.robot_exceptions import (
     RobotRetrieveInspectionException,
 )
-from robot_interface.models.inspection import Audio, ThermalVideo, ThermalVideoMetadata
 from robot_interface.models.inspection.inspection import (
     AudioMetadata,
     Image,
     ImageMetadata,
     Video,
     VideoMetadata,
+    Audio,
+    ThermalVideo,
+    ThermalVideoMetadata,
 )
 from robot_interface.models.mission.task import (
     RecordAudio,