diff --git a/extract_HydroSig.py b/extract_HydroSig.py
index da029ec2ca0f7a66abb9acb61540b24f1c2a9c99..e0226f50af5994943db9739b0b67b9a6278532e3 100644
--- a/extract_HydroSig.py
+++ b/extract_HydroSig.py
@@ -35,15 +35,15 @@ shp_watersheds = gpd.read_file(shp_watersheds_path)
 # List of the stations'codes
 watershed_code = shp_watersheds.loc[:,'Code']
 # Get a sample of them for test
-# code_for_test = watershed_code.loc[0:3]
+code_for_test = watershed_code.loc[509:512]
 # code_for_test = ['V1015810']
 
 # *************************** Extract watersheds' properties ************************************
 
 # problematic_watersheds = pd.DataFrame()
 studied_watersheds = pd.DataFrame()
-# for i in code_for_test:
-for i in watershed_code:
+for i in code_for_test:
+# for i in watershed_code:
     # Get the station's name
     watershed_stations_i = shp_watersheds[shp_watersheds.loc[:, 'Code'] == i]
     watershed_name_i =  watershed_stations_i.loc[:, 'Nom'].values[0]