diff --git a/python_package/src/onlinemid/__init__.py b/python_package/src/onlinemid/__init__.py
index 35cb2f1a513d7170d02f1d4bf67328e4621e4d78..88c7ba3d2d640731be36e8349a794a4fbf34a114 100644
--- a/python_package/src/onlinemid/__init__.py
+++ b/python_package/src/onlinemid/__init__.py
@@ -4,6 +4,9 @@ from .utils.agipd import agipd_module_name_from_mid_agipd_source
 from .utils.agipd import mid_agipd_source_from_agipd_module_name
 from .utils.agipd import agipd_module_name_from_agipd_module_number
 from .utils.agipd import agipd_module_number_from_agipd_module_name
+from .utils.agipd import generate_test_data
+
 from .utils.hash_and_dict import dict_from_hash
 from .utils.hash_and_dict import hash_from_dict
+
 from .utils.online_data_structures import NpVec