fix main exit without two args

diff --git a/src/main.c b/src/main.c
index 5f0bb2fb28943817f56fd1e52a3142c37f445d52..c148befbcd2fb9e646b25be7fe6b6a28c2d2552e 100644 (file)
--- a/src/main.c
+++ b/src/main.c
@@ -77,18 +77,26 @@ int main(int argc, char* argv[])
        }
 
        const char* dir = argv[1];
-       if (argc == 2)
+       if (argc == 2) {
                rename_files(dir, rename_string);
+               return EXIT_SUCCESS;
+       }
 
        const char* cmd = argv[2];
-       if (strcmp(cmd, "lower") == 0)
+       if (strcmp(cmd, "lower") == 0) {
                rename_files(dir, rename_lower);
+               return EXIT_SUCCESS;
+       }
 
-       if (strcmp(cmd, "point") == 0)
+       if (strcmp(cmd, "point") == 0) {
                rename_files(dir, rename_point);
+               return EXIT_SUCCESS;
+       }
 
-       if (strcmp(cmd, "revert") == 0)
+       if (strcmp(cmd, "revert") == 0) {
                rename_files(dir, rename_revert);
+               return EXIT_SUCCESS;
+       }
 
        return EXIT_SUCCESS;
 }
\ No newline at end of file