diff --git a/ansible/opnsense/update_opnsense_ha_cluster.yml b/ansible/opnsense/update_opnsense_ha_cluster.yml
index a94ab7d..6db0d4d 100644
--- a/ansible/opnsense/update_opnsense_ha_cluster.yml
+++ b/ansible/opnsense/update_opnsense_ha_cluster.yml
@@ -247,6 +247,8 @@
 
         - name: Take Proxmox VM snapshot
           community.general.proxmox_snap:
+            api_token_id: "{{ lookup('env', 'PROXMOX_TOKEN_ID') }}"
+            api_token_secret: "{{ lookup('env', 'PROXMOX_TOKEN_SECRET') }}"
             vmid: "{{ proxmox_vmid }}"
             state: present
             snapname: "{{ proxmox_snap_name }}"
@@ -348,6 +350,7 @@
             vmid: "{{ proxmox_vmid }}"
             state: rollback
             snapname: "{{ proxmox_snap_name }}"
+          register: _proxmox_snapshot_rollback
           ignore_errors: true
 
         - name: Send Ntfy notification
@@ -360,6 +363,7 @@
             body: |
               Current version: {{ firmware_current_version }}
               Update to {{ firmware_new_version }} failed on {{ inventory_hostname }} ({{ main_role }}).
+              {% if }
               A snapshot have been taken before the update and the VM has been rollback.
             headers:
               Title: "OPNsense update failed on {{ inventory_hostname }}"