diff --git a/bemade_geo_distance/models/bemade_geo_router.py b/bemade_geo_distance/models/bemade_geo_router.py
index 638d7ed..4a54392 100644
--- a/bemade_geo_distance/models/bemade_geo_router.py
+++ b/bemade_geo_distance/models/bemade_geo_router.py
@@ -29,7 +29,7 @@ class GeoRouter(models.AbstractModel):
 
     @api.model
     def get_driving_distance_time(self, origin, destination, departure_time: datetime.datetime = None,
-                                  arrival_time: datetime.date = None, units: units = 'metric',
+                                  arrival_time: datetime.datetime = None, units: units = 'metric',
                                   avoid_tolls=False, avoid_highways=False, avoid_ferries=False) -> (float, float):
         """ Calculates the route between two addresses and returns the distance it either in Kilometers or Miles and
             the time in minutes.
diff --git a/bemade_geo_distance/tests/test_geo_router.py b/bemade_geo_distance/tests/test_geo_router.py
index dd60f01..11de053 100644
--- a/bemade_geo_distance/tests/test_geo_router.py
+++ b/bemade_geo_distance/tests/test_geo_router.py
@@ -26,7 +26,7 @@ class TestGeoRouter(TransactionCase):
         arrival_time_utc = arrival_time_local.astimezone(pytz.utc)
         arrival_time = arrival_time_utc.strftime('%Y-%m-%dT%H:%M:%S.%fZ')
         print(arrival_time)
-        
+
         distance, time = self.env['base.geo_router'].get_driving_distance_time(bemade, montreal,
                                                                                arrival_time=arrival_time)