Makefile: Updated scripts to run tests in travis

1 files changed, 3 insertions, 2 deletions

diff --git a/input_fail.py b/input_fail.py
index 4b7e802..7bf33c3 100755
--- a/input_fail.py
+++ b/input_fail.py
@@ -1,13 +1,14 @@
 #!/usr/bin/env python
 
 # Exit with 1 if any input is provided. Print the input to stdout, unless an
-# argument is specified. In that case, also print the argument.
+# argument is specified. In that case, print the argument instead.
 
 import sys
 
 input_string = sys.stdin.read()
 if input_string != "":
-    sys.stdout.write(input_string)
     if len(sys.argv) >= 2:
         print(sys.argv[1])
+    else:
+        sys.stdout.write(input_string)
     sys.exit(1)