diff --git a/main.sh b/main.sh
index c16dbc5..6e15928 100755
--- a/main.sh
+++ b/main.sh
@@ -3,6 +3,7 @@
 set -e
 
 playbook="$INPUT_PLAYBOOK"
+directory="$INPUT_DIRECTORY"
 key="$INPUT_KEY"
 options="$INPUT_OPTIONS"
 
@@ -11,11 +12,18 @@ if test -z "$playbook"; then
     exit 1
 fi
 
+if test -z "$directory"; then
+    echo "You need to specify 'directory' input (root Ansible project directory)"
+    exit 1
+fi
+
 if test -z "$key"; then
     echo "You need to specify 'key' input (SSH private key)"
     exit 1
 fi
 
+cd "$directory"
+
 mkdir -p "$HOME/.ssh"
 echo "$key" > "$HOME/.ssh/id_rsa"
 chmod 600 "$HOME/.ssh/id_rsa"