diff --git a/customize-WebUI.py b/customize-WebUI.py
index d5df441..098e583 100644
--- a/customize-WebUI.py
+++ b/customize-WebUI.py
@@ -252,7 +252,7 @@ def main():
         print("Errors occurred during the process. Check the output for details.")
         print(f"{errorList['copies']} copy errors.")
         print(f"{errorList['modifications']} modification errors.")
-    elif errorList["warnings"] > 0:
+    elif errorList["copyWarnings"] > 0 or errorList["modWarnings"] > 0:
         print(f"All operations in {destinationDirectory} from {configPath} completed successfully! But there were {errorList['modWarnings'] + errorList['copyWarnings']} warnings. Mybe you should check them.")
     else:
         print(f"All operations in {destinationDirectory} from {configPath} completed successfully!")