diff --git a/tasks/main.yml b/tasks/main.yml
index 99d8d84..8aafb8a 100644
--- a/tasks/main.yml
+++ b/tasks/main.yml
@@ -3,7 +3,7 @@
 
 - name: Obtain server registration report
   opdk_server_registration_state:
-    server_type: '{{ server_type }}'
+    server_self: '{{ server_self }}'
     username: '{{ opdk_user_email }}'
     password: '{{ opdk_user_pass }}'
     mgmt_server_ip: '{{ local_mgmt_ip }}'
diff --git a/tests/test.yml b/tests/test.yml
index 31feec2..3d29be4 100644
--- a/tests/test.yml
+++ b/tests/test.yml
@@ -3,4 +3,4 @@
 - hosts: pgstandby
   gather_facts: no
   roles:
-    - { role: apigee-opdk-server-registration-state, username: "{{ opdk_user_email }}", password: "{{ opdk_user_pass }}", server_type: "{{ edge_ps_self }}", mgmt_server_ip: "{{ local_mgmt_ip }}" }
\ No newline at end of file
+    - { role: apigee-opdk-server-registration-state, username: "{{ opdk_user_email }}", password: "{{ opdk_user_pass }}", server_self: "{{ edge_ps_self }}", mgmt_server_ip: "{{ local_mgmt_ip }}" }
\ No newline at end of file