diff --git a/mu_map/data/datasets.py b/mu_map/data/datasets.py
index d0ae39c45aeab9744956d213dd3196c31d740b47..e92ad5acaa1a92f5877fdc36867256d8f9615b24 100644
--- a/mu_map/data/datasets.py
+++ b/mu_map/data/datasets.py
@@ -1,6 +1,7 @@
 import os
 from typing import Optional
 
+import cv2 as cv
 import pandas as pd
 import pydicom
 import numpy as np
@@ -129,8 +130,6 @@ __all__ = [MuMapDataset.__name__]
 if __name__ == "__main__":
     import argparse
 
-    import cv2 as cv
-
     from mu_map.util import to_grayscale, COLOR_WHITE
 
     parser = argparse.ArgumentParser(