diff --git a/roles/container/tasks/proxy.yml b/roles/container/tasks/proxy.yml
index dd9daa3..e59324f 100644
--- a/roles/container/tasks/proxy.yml
+++ b/roles/container/tasks/proxy.yml
@@ -1,5 +1,4 @@
 ---
-
 - name: Reverse proxy
   include_role:
     name: uumas.general.vhost
@@ -8,7 +7,7 @@
     vhost_id: "{{ docker_service_name }}"
     vhost_proxy_target_protocol: "{{ docker_proxy_target_protocol }}"
     vhost_domains: "{{ docker_vhost_domains[docker_service_name] }}"
-    vhost_proxy_target_port: "{{ ports[docker_service_name][proxy_target_protocol] }}"
+    vhost_proxy_target_port: "{{ ports[docker_service_name][vhost_proxy_target_protocol] }}"
   when: reverse_proxy_type != 'traefik'
 
 - name: Set published ports variable to http port
diff --git a/roles/service/tasks/proxy.yml b/roles/service/tasks/proxy.yml
index 687a755..e8de601 100644
--- a/roles/service/tasks/proxy.yml
+++ b/roles/service/tasks/proxy.yml
@@ -16,7 +16,7 @@
     vhost_id: "{{ docker_service_name }}"
     vhost_proxy_target_protocol: "{{ docker_proxy_target_protocol }}"
     vhost_domains: "{{ docker_vhost_domains[docker_service_name] }}"
-    vhost_proxy_target_port: "{{ ports[docker_service_name][proxy_target_protocol] }}"
+    vhost_proxy_target_port: "{{ ports[docker_service_name][vhost_proxy_target_protocol] }}"
   when: reverse_proxy_type != 'traefik'
 
 - name: Set published ports variable to http port