diff --git a/main.sh b/main.sh
index 1373268..2c7f482 100755
--- a/main.sh
+++ b/main.sh
@@ -1,7 +1,7 @@
 #!/bin/sh
 # WIP: Do not use this at the moment
 if [ -e "main" ]; then
-	./main >> /var/log/rawleyxyz/log 2>&1
+	$1/main >> /var/log/rawleyxyz/log 2>&1
 else
-	go build "main.go" && ./main &> /var/log/rawleyxyz/log
+	go build "$1/main.go" && $1/main &> /var/log/rawleyxyz/log
 fi