diff --git a/server/main.tf b/server/main.tf
index e13c8bd..9a32e5b 100644
--- a/server/main.tf
+++ b/server/main.tf
@@ -8,7 +8,7 @@ variable "identity" {}
 
 resource "aws_key_pair" "training" {
   key_name   = "${var.identity}-key"
-  public_key = "${file("~/.ssh/id_rsa.pub")}"
+  public_key = "${var.public_key}"
 }
 
 resource "aws_instance" "web" {
@@ -29,7 +29,7 @@ resource "aws_instance" "web" {
 
   connection {
     user        = "ubuntu"
-    private_key = "${file("~/.ssh/id_rsa")}"
+    private_key = "${var.private_key}"
   }
 
   provisioner "file" {