diff --git a/Tests/TestCase.php b/Tests/TestCase.php
index 732141d..5cca30a 100644
--- a/Tests/TestCase.php
+++ b/Tests/TestCase.php
@@ -195,7 +195,7 @@ public function getContainer()
     /**
      * Returns the an application mock which returns a mocked kernel
      *
-     * @return Application
+     * @return \Mockery\MockInterface|Application
      */
     protected function getApplication()
     {
@@ -282,7 +282,7 @@ protected function generateMigrationClassFileFromTemplate(array $variables)
     /**
      * Returns the version number from a string
      *
-     * @param $string
+     * @param string $string
      * @return string
      *
      * @throws \RuntimeException