diff --git a/src/View/Helper/CrudViewHelper.php b/src/View/Helper/CrudViewHelper.php
index 62e0142..3dafdc5 100644
--- a/src/View/Helper/CrudViewHelper.php
+++ b/src/View/Helper/CrudViewHelper.php
@@ -3,8 +3,8 @@
 
 namespace CrudView\View\Helper;
 
-use Cake\Database\Schema\TableSchemaInterface;
 use Cake\Datasource\EntityInterface;
+use Cake\Datasource\SchemaInterface;
 use Cake\Utility\Inflector;
 use Cake\Utility\Text;
 use Cake\View\Helper;
@@ -388,9 +388,9 @@ public function currentModel(): string
     /**
      * Get model schema.
      *
-     * @return \Cake\Database\Schema\TableSchemaInterface
+     * @return \Cake\Datasource\SchemaInterface
      */
-    public function schema(): TableSchemaInterface
+    public function schema(): SchemaInterface
     {
         return $this->getViewVar('modelSchema');
     }