diff --git a/src/LanguageLine.php b/src/LanguageLine.php
index 2ec66f9..f9d91a7 100644
--- a/src/LanguageLine.php
+++ b/src/LanguageLine.php
@@ -21,7 +21,7 @@ public static function boot()
     {
         parent::boot();
 
-        $flushGroupCache = function (LanguageLine $languageLine) {
+        $flushGroupCache = function (self $languageLine) {
             $languageLine->flushGroupCache();
         };
 
@@ -35,7 +35,7 @@ public static function getTranslationsForGroup(string $locale, string $group): a
             return static::query()
                 ->where('group', $group)
                 ->get()
-                ->reduce(function ($lines, LanguageLine $languageLine) use ($locale) {
+                ->reduce(function ($lines, self $languageLine) use ($locale) {
                     $translation = $languageLine->getTranslation($locale);
                     if ($translation !== null) {
                         Arr::set($lines, $languageLine->key, $translation);
diff --git a/tests/TranslationLoaders/DbTest.php b/tests/TranslationLoaders/DbTest.php
index 2eaee43..099a61f 100644
--- a/tests/TranslationLoaders/DbTest.php
+++ b/tests/TranslationLoaders/DbTest.php
@@ -3,7 +3,6 @@
 namespace Spatie\TranslationLoader\Test\TranslationLoaders;
 
 use DB;
-use Illuminate\Translation\Translator;
 use Spatie\TranslationLoader\LanguageLine;
 use Spatie\TranslationLoader\Test\TestCase;
 use Spatie\TranslationLoader\Exceptions\InvalidConfiguration;