Better output for input_fail.py

1 files changed, 2 insertions, 3 deletions

diff --git a/input_fail.py b/input_fail.py
index 67f5e74..4b7e802 100755
--- a/input_fail.py
+++ b/input_fail.py
@@ -1,14 +1,13 @@
 #!/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, print the argument.
+# argument is specified. In that case, also print the argument.
 
 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)