diff --git a/.github/workflows/unit_tests.sh b/.github/workflows/unit_tests.sh
index a5b98e089b..dc5bf5dd4f 100755
--- a/.github/workflows/unit_tests.sh
+++ b/.github/workflows/unit_tests.sh
@@ -22,6 +22,7 @@ ADDITIONAL_DEPS=(
     python3-pefile
     python3-pyelftools
     python3-pyparsing
+    python3-pytest
     rpm
     zstd
 )
diff --git a/src/ukify/ukify.py b/src/ukify/ukify.py
index 6e9d86b783..98739969ae 100755
--- a/src/ukify/ukify.py
+++ b/src/ukify/ukify.py
@@ -871,7 +871,7 @@ def generate_key_cert_pair(
     # supported/expected:
     # https://learn.microsoft.com/en-us/windows-hardware/manufacture/desktop/windows-secure-boot-key-creation-and-management-guidance?view=windows-11#12-public-key-cryptography
 
-    now = datetime.datetime.now(datetime.UTC)
+    now = datetime.datetime.now(datetime.timezone.utc)
 
     key = rsa.generate_private_key(
         public_exponent=65537,