diff --git a/src/cal_tools/agipdlib.py b/src/cal_tools/agipdlib.py
index 03ea731d88a4632f0f99925069869fa845fbfd16..3ae30fafb4df92e44778f714de9c97c5140a5297 100644
--- a/src/cal_tools/agipdlib.py
+++ b/src/cal_tools/agipdlib.py
@@ -3,6 +3,7 @@ import posixpath
 import zlib
 from dataclasses import dataclass, field
 from datetime import datetime
+from logging import warning
 from multiprocessing import Manager
 from multiprocessing.pool import ThreadPool
 from typing import List, Optional
@@ -11,8 +12,7 @@ import h5py
 import numpy as np
 import sharedmem
 from dateutil import parser
-from extra_data import DataCollection, H5File, by_id, components
-from logging import warning
+from extra_data import DataCollection, H5File, RunDirectory, by_id, components
 
 from cal_tools import agipdalgs as calgs
 from cal_tools.agipdutils import (
@@ -292,7 +292,7 @@ class AgipdCtrl:
 
         return 12
 
-from extra_data import RunDirectory
+
 @dataclass
 class AgipdCtrlRuns:
     raw_folder: str