diff --git a/tests/block/template.php b/tests/block/template.php
index 4ff329f..4a92b78 100644
--- a/tests/block/template.php
+++ b/tests/block/template.php
@@ -1,6 +1,6 @@
 <?php
 /**
- * @var array{class_name: string, content: string} $args;
+ * @var array{class_name: string, content: string, wp_block: WP_Block} $args;
  */
 ?>
 <div class="<?php echo esc_attr( $args['class_name'] ); ?>"><?php echo esc_html( $args['content'] ); ?></div>