diff --git a/src/Database/DatabaseMigrationRepository.php b/src/Database/DatabaseMigrationRepository.php
index 3e02ae84a..5ba06cc09 100644
--- a/src/Database/DatabaseMigrationRepository.php
+++ b/src/Database/DatabaseMigrationRepository.php
@@ -51,6 +51,7 @@ class DatabaseMigrationRepository implements MigrationRepositoryInterface
     /**
      * Get the ran migrations.
      *
+     * @param string $extension
      * @return array
      */
     public function getRan($extension = null)
diff --git a/src/Database/MigrationRepositoryInterface.php b/src/Database/MigrationRepositoryInterface.php
index fccc76ac2..7bd9265e8 100644
--- a/src/Database/MigrationRepositoryInterface.php
+++ b/src/Database/MigrationRepositoryInterface.php
@@ -16,6 +16,7 @@ interface MigrationRepositoryInterface
     /**
      * Get the ran migrations for the given extension.
      *
+     * @param string $extension
      * @return array
      */
     public function getRan($extension = null);