diff --git a/app/Policies/TokenPolicy.php b/app/Policies/TokenPolicy.php
index 36e37e313a69..fff0ffc1f6e6 100644
--- a/app/Policies/TokenPolicy.php
+++ b/app/Policies/TokenPolicy.php
@@ -2,12 +2,14 @@
 
 namespace App\Policies;
 
+use App\Models\User;
+
 class TokenPolicy extends EntityPolicy {
-    public static function edit($user, $item) {
+    public static function edit(User $user, $item) {
         return $user->hasPermission('admin');
     }
 
-    public static function create($user) {
+    public static function create(User $user) {
         return $user->hasPermission('admin');
     }
-}
\ No newline at end of file
+}
diff --git a/app/Providers/AuthServiceProvider.php b/app/Providers/AuthServiceProvider.php
index bd826f64056c..b395cca2c668 100644
--- a/app/Providers/AuthServiceProvider.php
+++ b/app/Providers/AuthServiceProvider.php
@@ -25,7 +25,7 @@ class AuthServiceProvider extends ServiceProvider
         \App\Models\Product::class => \App\Policies\ProductPolicy::class,
         \App\Models\TaxRate::class => \App\Policies\TaxRatePolicy::class,
         \App\Models\AccountGateway::class => \App\Policies\AccountGatewayPolicy::class,
-        \App\Models\Token::class => \App\Policies\TokenPolicy::class,
+        \App\Models\AccountToken::class => \App\Policies\TokenPolicy::class,
         \App\Models\BankAccount::class => \App\Policies\BankAccountPolicy::class,
         \App\Models\PaymentTerm::class => \App\Policies\PaymentTermPolicy::class,
     ];