diff --git a/src/TestSessionController.php b/src/TestSessionController.php
index b6933a6..ff91357 100644
--- a/src/TestSessionController.php
+++ b/src/TestSessionController.php
@@ -361,7 +361,7 @@ class TestSessionController extends Controller
     }
 
     /**
-     * @return ArrayList
+     * @return ArrayList<ArrayData>
      */
     public function getState()
     {