1 files changed, 7 insertions, 8 deletions
diff --git a/test/units/module_utils/facts/test_date_time.py b/test/units/module_utils/facts/test_date_time.py
index 6abc36a7..6cc05f97 100644
--- a/test/units/module_utils/facts/test_date_time.py
+++ b/test/units/module_utils/facts/test_date_time.py
@@ -10,28 +10,27 @@ import datetime
 import string
 import time
 
+from ansible.module_utils.compat.datetime import UTC
 from ansible.module_utils.facts.system import date_time
 
 EPOCH_TS = 1594449296.123456
 DT = datetime.datetime(2020, 7, 11, 12, 34, 56, 124356)
-DT_UTC = datetime.datetime(2020, 7, 11, 2, 34, 56, 124356)
+UTC_DT = datetime.datetime(2020, 7, 11, 2, 34, 56, 124356)
 
 
 @pytest.fixture
 def fake_now(monkeypatch):
     """
-    Patch `datetime.datetime.fromtimestamp()`, `datetime.datetime.utcfromtimestamp()`,
+    Patch `datetime.datetime.fromtimestamp()`,
     and `time.time()` to return deterministic values.
     """
 
     class FakeNow:
         @classmethod
-        def fromtimestamp(cls, timestamp):
-            return DT
-
-        @classmethod
-        def utcfromtimestamp(cls, timestamp):
-            return DT_UTC
+        def fromtimestamp(cls, timestamp, tz=None):
+            if tz == UTC:
+                return UTC_DT.replace(tzinfo=tz)
+            return DT.replace(tzinfo=tz)
 
     def _time():
         return EPOCH_TS