diff --git a/public/partials/confirmation-screens.php b/public/partials/confirmation-screens.php
index 10e02252..48e715e1 100644
--- a/public/partials/confirmation-screens.php
+++ b/public/partials/confirmation-screens.php
@@ -55,7 +55,7 @@
 			<p><?php echo esc_html( $text ); ?></p>
 		</div>
 		<footer>
-			<button class="gdpr-ok"><?php esc_html_e( 'OK', 'gdpr' ); ?></button>
+			<button class="gdpr-ok" data-callback="closeNotification"><?php esc_html_e( 'OK', 'gdpr' ); ?></button>
 		</footer>
 	</div>
 </div>