diff --git a/src/hydroshoot/energy.py b/src/hydroshoot/energy.py
index 605a3d0..9c7be43 100644
--- a/src/hydroshoot/energy.py
+++ b/src/hydroshoot/energy.py
@@ -127,7 +127,7 @@ def set_leaf_temperature_to_air_temperature(air_temperature, leaf_ids):
     return {vid: air_temperature for vid in leaf_ids}
 
 
-def set_wind_speed(g, meteo, leaf_lbl_prefix='L'):
+def set_local_wind_speed(g, meteo, leaf_lbl_prefix='L') -> dict:
     """Basic model for wind speed at leaf level, considered equal to air wind speed for all leaves
 
     Args:
diff --git a/src/hydroshoot/initialisation.py b/src/hydroshoot/initialisation.py
index cc971f4..e36bfc0 100644
--- a/src/hydroshoot/initialisation.py
+++ b/src/hydroshoot/initialisation.py
@@ -8,7 +8,7 @@
 
 from hydroshoot import soil
 from hydroshoot.architecture import get_mtg_base, add_soil_surface_mesh, get_leaves
-from hydroshoot.energy import set_form_factors_simplified, set_wind_speed, set_local_air_temperature
+from hydroshoot.energy import set_form_factors_simplified, set_local_wind_speed, set_local_air_temperature
 from hydroshoot.exchange import leaf_Na
 from hydroshoot.io import HydroShootInputs, HydroShootHourlyInputs
 from hydroshoot.irradiance import irradiance_distribution, hsCaribu, set_optical_properties
@@ -169,8 +169,8 @@ def init_hourly(g: MTG, inputs_hourly: HydroShootHourlyInputs, leaf_ppfd: dict,
     # Add a date index to g
     g.date = datetime.strftime(inputs_hourly.date, "%Y%m%d%H%M%S")
 
-    # initiate wind speed
-    g.properties()['u'] = set_wind_speed(
+    # initiate local wind speed
+    g.properties()['u'] = set_local_wind_speed(
         g=g, meteo=inputs_hourly.weather, leaf_lbl_prefix=params.mtg_api.leaf_lbl_prefix)
 
     # initiate local air temperature