Test banner written

2 files changed, 5 insertions, 4 deletions

diff --git a/banner.c b/banner.c
index f774fe5..7541618 100644
--- a/banner.c
+++ b/banner.c
@@ -2,7 +2,5 @@
 
 /* Print the welcome and warning banners */
 int banner(void) {
-    int bw;
-    bw = fprintf(stdout, "%s\n%s\n", WELCOME, WARNING);
-    return (bw > 0);
+    return fprintf(stdout, "%s\n%s\n", WELCOME, WARNING);
 }
diff --git a/main.c b/main.c
index d50ca46..81030d6 100644
--- a/main.c
+++ b/main.c
@@ -4,7 +4,10 @@
 int main(void) {
 
     /* Show the banner with the warning */
-    banner();
+    if (banner() < 0) {
+        fputs("Couldn't write welcome banner to stdout!", stderr);
+        exit(EXIT_FAILURE);
+    }
 
     /* Start looping through commands */
     loop();