diff --git a/src/bactos_functions.py b/src/bactos_functions.py
index ca9f8a2b23f6dec9f549415df820523234cf37d0..2200a21ad10cc74a36a3bb6554d7d8cb31bb2734 100755
--- a/src/bactos_functions.py
+++ b/src/bactos_functions.py
@@ -1123,7 +1123,7 @@ def run_processing(data_list, nom_sortie, MM_version=2, background='auto',
                    stardist_path = None, bact_size=(0.5, 15), prob_thresh=None, 
                    time_to_extract_bacteria=0, build_bacteria_figures=True, 
                    bacteria_extract_mask='TRP', drift_algo='cor', register_filter=True, 
-                   register_stack=True, register_mask=True):
+                   register_stack=True, register_mask=True, drift_filter=0):
     """
     Loop over the given folder name to extract data and export them inside one hdf file (containing all data from each folder)
 
@@ -1178,6 +1178,9 @@ def run_processing(data_list, nom_sortie, MM_version=2, background='auto',
     - register_mask, bool:
         Register the two filter based on bacteria mask (default True)
         
+    - drift_filter, bool:
+        The filter on which to register the images in time
+        
     Example:
     --------
 
@@ -1232,7 +1235,7 @@ def run_processing(data_list, nom_sortie, MM_version=2, background='auto',
         print('stabilisation')
         imgs = register_images(imgs, gaussian_size=0, move_filter=drug_filter,
                                ref_filter=ctrl_filter, register_filter=register_filter,
-                               register_stack=register_stack, show_info=False, up_factor=10, drift_filter=0,
+                               register_stack=register_stack, show_info=False, up_factor=10, drift_filter=drift_filter,
                                scale_correction=scale_correction, 
                                drift_algo=drift_algo)