diff --git a/helpers/state_machine.sh b/helpers/state_machine.sh
index 19b5033..fa74f27 100755
--- a/helpers/state_machine.sh
+++ b/helpers/state_machine.sh
@@ -15,7 +15,7 @@ set -o allexport
 source ${script_dir}/../config/state_machine
 set +o allexport
 
-mkdir -p ${script_dir}/storage/state_machine
+mkdir -p ${script_dir}/../storage/state_machine
 
 # Get input from standard input or via first parameter
 if [[ $# -eq 0 ]]; then
@@ -38,7 +38,7 @@ if [[ -z "${key}" ]]; then
   exit 1
 fi
 
-key_file="${script_dir}/storage/state_machine/${key}.txt"
+key_file="${script_dir}/../storage/state_machine/${key}.txt"
 
 if [[ -f $key_file && -z "${message}" ]]; then
   # Previous message present and empty message now