diff --git a/components/i_feed/test_ifeed.py b/components/i_feed/test_ifeed.py
index 2d8b393..d6e79f0 100644
--- a/components/i_feed/test_ifeed.py
+++ b/components/i_feed/test_ifeed.py
@@ -47,9 +47,8 @@ class TestVehicle(unittest.TestCase):
             self.assertEqual(self.v.nce, False)
 
     def test_daf_prop(self):
-        with patch('datetime.datetime', MyDate):
-            # noinspection PyUnresolvedReferences
-            datetime.datetime.set_timestamp(self.timestamp)
+        with patch('devices.vehicle.datetime', MyDate):
+            MyDate.set_timestamp(self.timestamp)
 
             # test if daf object created properly
             daf = DAF(vehicle_identification_number=self.vin,
@@ -60,13 +59,8 @@ class TestVehicle(unittest.TestCase):
             self.assertEqual(self.v.daf, daf)
 
     def test_current_location_prop(self):
-        with patch('datetime.datetime', MyDate):
-            # noinspection PyUnresolvedReferences
-            datetime.datetime.set_timestamp(self.timestamp)
-
-            # assert timestamp mocking worked
-            self.assertEqual(type(datetime.datetime), type(MyDate))
-            self.assertEqual(self.timestamp, datetime.datetime.now())
+        with patch('devices.vehicle.datetime', MyDate):
+            MyDate.set_timestamp(self.timestamp)
 
             # initial position should be starting point
             self.assertEqual(self.v.gps_location, self.starting_point)