diff --git a/src/Appwrite/Migration/Migration.php b/src/Appwrite/Migration/Migration.php
index f2d6a8c0a5..8d8100ee74 100644
--- a/src/Appwrite/Migration/Migration.php
+++ b/src/Appwrite/Migration/Migration.php
@@ -26,7 +26,7 @@ abstract class Migration
   /**
    * Set project for migration.
    */
-  public function setProject(Document $project, Database $projectDB)
+  public function setProject(Document $project, Database $projectDB): Migration
   {
     $this->project = $project;
     $this->projectDB = $projectDB;