Fix program_name in usage statement

1 files changed, 6 insertions, 3 deletions

diff --git a/main.c b/main.c
index 34d4201..4ab8bb6 100644
--- a/main.c
+++ b/main.c
@@ -29,11 +29,14 @@ const char *USAGE_STATEMENT = \
 
 void usage() {
     char *name;
+    unsigned is_base;
+
+    is_base = 0;
     name = strrchr(program_name, '/');
-    if (name == NULL) {
-        name = program_name;
+    if (name != NULL) {
+        is_base = 1;
     }
-    printf(USAGE_STATEMENT, name + 1, VERSION);
+    printf(USAGE_STATEMENT, is_base ? name + 1 : program_name, VERSION);
 }
 
 int edit_file(const char *filename) {