diff --git a/src/IP.php b/src/IP.php
index 162b9c1..9a607c3 100644
--- a/src/IP.php
+++ b/src/IP.php
@@ -2,8 +2,8 @@
 
 namespace Mchev\Banhammer;
 
-use Illuminate\Support\Facades\Cache;
 use Illuminate\Database\Eloquent\Builder;
+use Illuminate\Support\Facades\Cache;
 use Mchev\Banhammer\Models\Ban;
 
 class IP
@@ -44,7 +44,7 @@ public static function banned(): Builder
             ->groupBy('ip');
     }
 
-    public static function getBannedIPsFromCache(): Array
+    public static function getBannedIPsFromCache(): array
     {
         return Cache::has('banned-ips')
             ? Cache::get('banned-ips')
diff --git a/src/Middleware/IPBanned.php b/src/Middleware/IPBanned.php
index a18e2b6..6818b0b 100644
--- a/src/Middleware/IPBanned.php
+++ b/src/Middleware/IPBanned.php
@@ -3,7 +3,7 @@
 namespace Mchev\Banhammer\Middleware;
 
 use Closure;
-use Illuminate\Support\Facades\Cache;
+use Mchev\Banhammer\IP;
 use Symfony\Component\HttpFoundation\Response;
 
 class IPBanned
diff --git a/src/Middleware/LogoutBanned.php b/src/Middleware/LogoutBanned.php
index a2320be..f339b7e 100644
--- a/src/Middleware/LogoutBanned.php
+++ b/src/Middleware/LogoutBanned.php
@@ -3,7 +3,7 @@
 namespace Mchev\Banhammer\Middleware;
 
 use Closure;
-use Illuminate\Support\Facades\Cache;
+use Mchev\Banhammer\IP;
 use Symfony\Component\HttpFoundation\Response;
 
 class LogoutBanned