diff --git a/6/1/solution.sh b/6/1/solution.sh
index 554b069..d84bedc 100755
--- a/6/1/solution.sh
+++ b/6/1/solution.sh
@@ -7,12 +7,12 @@ if [[ "${TRACE-0}" == "1" ]]; then
   set -o xtrace
 fi
 
-cd "$(dirname "$0")"
+INPUT="$1"
 
 main ()
 {
-  IFS=' ' read -r -a times <<< "$(grep Time input | awk '{$1=""; print $0}')"
-  IFS=' ' read -r -a distances <<< "$(grep Distance input | awk '{$1=""; print $0}')"
+  IFS=' ' read -r -a times <<< "$(grep Time ${INPUT} | awk '{$1=""; print $0}')"
+  IFS=' ' read -r -a distances <<< "$(grep Distance ${INPUT} | awk '{$1=""; print $0}')"
   declare -a winning_counts=()
   records="${#times[@]}"
   for ((i=0; i<records; i++)); do