diff --git a/ci_tests/test_pygoat_findings.py b/ci_tests/test_pygoat_findings.py
index 7602758f..ea3c6dfb 100644
--- a/ci_tests/test_pygoat_findings.py
+++ b/ci_tests/test_pygoat_findings.py
@@ -14,6 +14,7 @@
     "pixee:python/url-sandbox",
     "pixee:python/use-defusedxml",
     "pixee:python/use-walrus-if",
+    "pixee:python/timezone-aware-datetime",
 ]
 
 
diff --git a/src/core_codemods/timezone_aware_datetime.py b/src/core_codemods/timezone_aware_datetime.py
index d3d84592..bca4103a 100644
--- a/src/core_codemods/timezone_aware_datetime.py
+++ b/src/core_codemods/timezone_aware_datetime.py
@@ -74,9 +74,8 @@ def leave_Call(self, original_node: cst.Call, updated_node: cst.Call):
         return updated_node
 
     def _determine_module_and_kwarg(self, original_node: cst.Call):
-        maybe_name = self.get_aliased_prefix_name(original_node, self._module_name)
 
-        if maybe_name:
+        if maybe_name := self.get_aliased_prefix_name(original_node, self._module_name):
             # it's a regular import OR alias import
             if maybe_name == self._module_name:
                 module = "datetime.datetime"