diff --git a/energy_demand/geography/weather_station_location.py b/energy_demand/geography/weather_station_location.py
index 84e87f02..dde7c90e 100644
--- a/energy_demand/geography/weather_station_location.py
+++ b/energy_demand/geography/weather_station_location.py
@@ -26,7 +26,7 @@ def calc_distance_two_points(lat_from, long_from, lat_to, long_to):
     distance_in_km = haversine(
         (lat_from, long_from),
         (lat_to, long_to),
-        miles=False)
+        unit='km')
 
     return distance_in_km