diff --git a/scripts/db_migrator.py b/scripts/db_migrator.py
index fc18eb56e2..1a2041a637 100755
--- a/scripts/db_migrator.py
+++ b/scripts/db_migrator.py
@@ -1162,7 +1162,7 @@ def migrate(self):
             yang_parser.validate_data_tree()
         except sonic_yang.SonicYangException as e:
             syslog.syslog(syslog.LOG_CRIT, "Yang validation failed: " + str(e))
-            if os.environ["UTILITIES_UNIT_TESTING"] == "2":
+            if os.environ.get("UTILITIES_UNIT_TESTING", "0") == "2":
                 raise