diff --git a/image_processing/build_dataset.py b/image_processing/build_dataset.py
index 1c1b6584f4a7eb14477b6f3af77bf3fc1c5a27a7..67718f795a80d675e817fe5940ceae8e532bd195 100644
--- a/image_processing/build_dataset.py
+++ b/image_processing/build_dataset.py
@@ -7,7 +7,7 @@ import numpy as np
 import matplotlib.image as mpimg
 
 
-from image_processing.build_image import build_image_ms1_wiff, build_image_ms1_wiff_charge_filtered, build_image_ms1_wiff_charge_filtered_apex_only
+from build_image import build_image_ms1_wiff_charge_filtered, build_image_ms1_wiff_charge_filtered_apex_only
 
 """
 find . -name '*.mzML' -exec cp -prv '{}' '/home/leo/PycharmProjects/pseudo_image/data/raw_data' ';'