diff --git a/Win11DisableRoundedCorners/Win11DisableRoundedCorners.c b/Win11DisableRoundedCorners/Win11DisableRoundedCorners.c
index a2aa088..ded8753 100644
--- a/Win11DisableRoundedCorners/Win11DisableRoundedCorners.c
+++ b/Win11DisableRoundedCorners/Win11DisableRoundedCorners.c
@@ -8,6 +8,11 @@ int main(int argc, char** argv)
 {
     BOOL bRestore = FALSE;
 
+    if (argc == 2 && strcmp(argv[1], "-restore") == 0)
+    {
+        bRestore = TRUE;
+    }
+
     char szTaskkill[MAX_PATH];
     ZeroMemory(
         szTaskkill,
@@ -50,7 +55,6 @@ int main(int argc, char** argv)
         MAX_PATH,
         "\\uDWM_win11drc.bak"
     );
-    bRestore = fileExists(szOriginalDWM);
 
     char szModifiedDWM[_MAX_PATH];
     ZeroMemory(
@@ -290,4 +294,4 @@ int main(int argc, char** argv)
     }
     printf("Operation successful.\n");
 	return 0;
-}
\ No newline at end of file
+}