diff --git a/tests/integration/targets/storage_box/tasks/test.yml b/tests/integration/targets/storage_box/tasks/test.yml
index 12e45b4..d16b6a0 100644
--- a/tests/integration/targets/storage_box/tasks/test.yml
+++ b/tests/integration/targets/storage_box/tasks/test.yml
@@ -218,17 +218,22 @@
     that:
       - result is changed
 
-# TODO: take snapshot and rollback
-# - name: Test rollback snapshot
-#   hetzner.hcloud.storage_box:
-#     name: "{{ hcloud_storage_box_name }}"
-#     snapshot: "TODO"
-#     state: rollback_snapshot
-#   register: result
-# - name: Verify rollback snapshot
-#   ansible.builtin.assert:
-#     that:
-#       - result is changed
+- name: Create snapshot
+  hetzner.hcloud.storage_box_snapshot:
+    storage_box: "{{ hcloud_storage_box_name }}"
+    description: test-rollback
+  register: _rollback_snapshot
+
+- name: Test rollback snapshot
+  hetzner.hcloud.storage_box:
+    name: "{{ hcloud_storage_box_name }}"
+    snapshot: "{{ _rollback_snapshot.hcloud_storage_box_snapshot.id }}"
+    state: rollback_snapshot
+  register: result
+- name: Verify rollback snapshot
+  ansible.builtin.assert:
+    that:
+      - result is changed
 
 - name: Test delete
   hetzner.hcloud.storage_box: