diff --git a/worker.py b/worker.py
index f66c51d..5187513 100644
--- a/worker.py
+++ b/worker.py
@@ -354,12 +354,12 @@ def run(config, provider, location, server_type):
         ssh_execute_command(server_ip, command)
 
     # Check if private key and preshared key are provided
-    if config["wireguard"]["private_key"]:
+    if config.get("wireguard", "private_key"):
         private_key = config["wireguard"]["private_key"]
     else:
         private_key = generate_private_key()
 
-    if config["wireguard"]["preshared_key"]:
+    if config.get("wireguard", "preshared_key"):
         preshared_key = config["wireguard"]["preshared_key"]
     else:
         preshared_key = generate_preshared_key()