diff --git a/algo b/algo
index eba3eb4..8eeb05a 100755
--- a/algo
+++ b/algo
@@ -105,10 +105,10 @@ Note: Make sure to use either your root key (recommended) or an IAM user with an
 
 
   read -p "
-Enter the local path to your SSH public key (~/.ssh/id_rsa.pub): " -r ssh_public_key_file
-ssh_public_key=${ssh_public_key_file:-$HOME/.ssh/id_rsa.pub}
+Enter the local path to your SSH public key (~/.ssh/id_rsa.pub): " -r ssh_public_key
+  ssh_public_key=${ssh_public_key:-$HOME/.ssh/id_rsa.pub}
 
-  read -p "
+read -p "
 Name the vpn server:
 [algo]: " -r aws_server_name
   aws_server_name=${aws_server_name:-algo}
@@ -157,7 +157,7 @@ Enter the local path to your credentials JSON file (https://support.google.com/c
 
   read -p "
 Enter the local path to your SSH public key (~/.ssh/id_rsa.pub): " -r ssh_public_key
-ssh_public_key=${ssh_public_key_file:-$HOME/.ssh/id_rsa.pub}
+  ssh_public_key=${ssh_public_key:-$HOME/.ssh/id_rsa.pub}
 
   read -p "
 Name the vpn server:
diff --git a/roles/cloud-ec2/tasks/main.yml b/roles/cloud-ec2/tasks/main.yml
index e2b0a65..4b8de61 100644
--- a/roles/cloud-ec2/tasks/main.yml
+++ b/roles/cloud-ec2/tasks/main.yml
@@ -66,6 +66,7 @@
 - name: Add new instance to host group
   add_host:
     hostname: "{{ item.public_ip }}"
+    ansible_ssh_private_key_file: "{{ ssh_public_key[:-4] }}"
     groupname: vpn-host
     ansible_ssh_user: ubuntu
     ansible_python_interpreter: "/usr/bin/python2.7"