diff --git a/src/Repository/SqliteRepository.php b/src/Repository/SqliteRepository.php
index cfca5cc..361c608 100644
--- a/src/Repository/SqliteRepository.php
+++ b/src/Repository/SqliteRepository.php
@@ -2,7 +2,7 @@
 
 namespace Ulmus\Repository;
 
-use Ulmus\{Repository, Entity};
+use Ulmus\{EntityCollection, Repository, Entity};
 
 class SqliteRepository extends Repository {
     use JsonConditionTrait;
@@ -19,7 +19,7 @@ class SqliteRepository extends Repository {
 
     }
 
-    public function listColumns(? string $table = null)
+    public function listColumns(? string $table = null) : EntityCollection
     {
         $table ??= $this->entityResolver->tableName();
         $this->showColumnsSqlQuery($table);