diff --git a/src/App/Limitations/LimitationManager.php b/src/App/Limitations/LimitationManager.php
deleted file mode 100644
index 732f7e44..00000000
--- a/src/App/Limitations/LimitationManager.php
+++ /dev/null
@@ -1,30 +0,0 @@
-<?php
-namespace App\Limitations;
-
-use Illuminate\Support\Manager;
-use App\Limitations\Engines\DefaultLimitationEngine;
-use App\Limitations\Engines\FixedBillingLimitationEngine;
-use App\Limitations\Engines\MeteredBillingLimitationEngine;
-
-class LimitationManager extends Manager
-{
-    public function getDefaultDriver(): string
-    {
-        return get_limitation_driver();
-    }
-
-    public function createDefaultDriver(): DefaultLimitationEngine
-    {
-        return new DefaultLimitationEngine();
-    }
-
-    public function createFixedDriver(): FixedBillingLimitationEngine
-    {
-        return new FixedBillingLimitationEngine();
-    }
-
-    public function createMeteredDriver(): MeteredBillingLimitationEngine
-    {
-        return new MeteredBillingLimitationEngine();
-    }
-}
diff --git a/src/App/Limitations/Engines/DefaultLimitationEngine.php b/src/App/Restrictions/Engines/DefaultRestrictionsEngine.php
similarity index 88%
rename from src/App/Limitations/Engines/DefaultLimitationEngine.php
rename to src/App/Restrictions/Engines/DefaultRestrictionsEngine.php
index 464c8873..3cad7935 100644
--- a/src/App/Limitations/Engines/DefaultLimitationEngine.php
+++ b/src/App/Restrictions/Engines/DefaultRestrictionsEngine.php
@@ -1,11 +1,11 @@
 <?php
-namespace App\Limitations\Engines;
+namespace App\Restrictions\Engines;
 
 use App\Users\Models\User;
-use App\Limitations\LimitationEngine;
+use App\Restrictions\RestrictionsEngine;
 use Domain\Teams\Actions\CheckMaxTeamMembersLimitAction;
 
-class DefaultLimitationEngine implements LimitationEngine
+class DefaultRestrictionsEngine implements RestrictionsEngine
 {
     public function canUpload(User $user, int $fileSize = 0): bool
     {
diff --git a/src/App/Limitations/Engines/FixedBillingLimitationEngine.php b/src/App/Restrictions/Engines/FixedBillingRestrictionsEngine.php
similarity index 86%
rename from src/App/Limitations/Engines/FixedBillingLimitationEngine.php
rename to src/App/Restrictions/Engines/FixedBillingRestrictionsEngine.php
index 943fe03a..b255ec98 100644
--- a/src/App/Limitations/Engines/FixedBillingLimitationEngine.php
+++ b/src/App/Restrictions/Engines/FixedBillingRestrictionsEngine.php
@@ -1,11 +1,11 @@
 <?php
-namespace App\Limitations\Engines;
+namespace App\Restrictions\Engines;
 
 use App\Users\Models\User;
-use App\Limitations\LimitationEngine;
+use App\Restrictions\RestrictionsEngine;
 use Domain\Teams\Actions\CheckMaxTeamMembersLimitAction;
 
-class FixedBillingLimitationEngine implements LimitationEngine
+class FixedBillingRestrictionsEngine implements RestrictionsEngine
 {
     public function canUpload(User $user, int $fileSize = 0): bool
     {
diff --git a/src/App/Limitations/Engines/MeteredBillingLimitationEngine.php b/src/App/Restrictions/Engines/MeteredBillingRestrictionsEngine.php
similarity index 86%
rename from src/App/Limitations/Engines/MeteredBillingLimitationEngine.php
rename to src/App/Restrictions/Engines/MeteredBillingRestrictionsEngine.php
index bac7420d..4bea5eef 100644
--- a/src/App/Limitations/Engines/MeteredBillingLimitationEngine.php
+++ b/src/App/Restrictions/Engines/MeteredBillingRestrictionsEngine.php
@@ -1,10 +1,10 @@
 <?php
-namespace App\Limitations\Engines;
+namespace App\Restrictions\Engines;
 
 use App\Users\Models\User;
-use App\Limitations\LimitationEngine;
+use App\Restrictions\RestrictionsEngine;
 
-class MeteredBillingLimitationEngine implements LimitationEngine
+class MeteredBillingRestrictionsEngine implements RestrictionsEngine
 {
     public function canUpload(User $user, int $fileSize = 0): bool
     {
diff --git a/src/App/Limitations/LimitationEngine.php b/src/App/Restrictions/RestrictionsEngine.php
similarity index 86%
rename from src/App/Limitations/LimitationEngine.php
rename to src/App/Restrictions/RestrictionsEngine.php
index d63ccbed..d91e5876 100644
--- a/src/App/Limitations/LimitationEngine.php
+++ b/src/App/Restrictions/RestrictionsEngine.php
@@ -1,9 +1,9 @@
 <?php
-namespace App\Limitations;
+namespace App\Restrictions;
 
 use App\Users\Models\User;
 
-interface LimitationEngine
+interface RestrictionsEngine
 {
     public function canUpload(User $user, int $fileSize = 0): bool;
 
diff --git a/src/App/Restrictions/RestrictionsManager.php b/src/App/Restrictions/RestrictionsManager.php
new file mode 100644
index 00000000..d175bccf
--- /dev/null
+++ b/src/App/Restrictions/RestrictionsManager.php
@@ -0,0 +1,30 @@
+<?php
+namespace App\Restrictions;
+
+use Illuminate\Support\Manager;
+use App\Restrictions\Engines\DefaultRestrictionsEngine;
+use App\Restrictions\Engines\FixedBillingRestrictionsEngine;
+use App\Restrictions\Engines\MeteredBillingRestrictionsEngine;
+
+class RestrictionsManager extends Manager
+{
+    public function getDefaultDriver(): string
+    {
+        return get_restriction_driver();
+    }
+
+    public function createDefaultDriver(): DefaultRestrictionsEngine
+    {
+        return new DefaultRestrictionsEngine();
+    }
+
+    public function createFixedDriver(): FixedBillingRestrictionsEngine
+    {
+        return new FixedBillingRestrictionsEngine();
+    }
+
+    public function createMeteredDriver(): MeteredBillingRestrictionsEngine
+    {
+        return new MeteredBillingRestrictionsEngine();
+    }
+}
diff --git a/src/App/Users/Models/User.php b/src/App/Users/Models/User.php
index f40ca8dd..066434c1 100644
--- a/src/App/Users/Models/User.php
+++ b/src/App/Users/Models/User.php
@@ -12,9 +12,9 @@ use Illuminate\Support\Facades\DB;
 use Database\Factories\UserFactory;
 use Domain\Settings\Models\Setting;
 use Kyslik\ColumnSortable\Sortable;
-use App\Limitations\LimitationManager;
 use Illuminate\Support\Facades\Storage;
 use Illuminate\Notifications\Notifiable;
+use App\Restrictions\RestrictionsManager;
 use App\Users\Notifications\ResetPassword;
 use Laravel\Fortify\TwoFactorAuthenticatable;
 use Illuminate\Contracts\Auth\MustVerifyEmail;
@@ -203,7 +203,7 @@ class User extends Authenticatable implements MustVerifyEmail
     public function __call($method, $parameters)
     {
         if (str_starts_with($method, 'can')) {
-            return resolve(LimitationManager::class)
+            return resolve(RestrictionsManager::class)
                 ->driver()
                 ->$method($this, ...$parameters);
         }
diff --git a/src/Domain/Files/Controllers/FileAccess/GetFileController.php b/src/Domain/Files/Controllers/FileAccess/GetFileController.php
index 977172b0..019b9a85 100644
--- a/src/Domain/Files/Controllers/FileAccess/GetFileController.php
+++ b/src/Domain/Files/Controllers/FileAccess/GetFileController.php
@@ -2,6 +2,7 @@
 namespace Domain\Files\Controllers\FileAccess;
 
 use Gate;
+use Illuminate\Http\Response;
 use App\Http\Controllers\Controller;
 use Domain\Files\Models\File as UserFile;
 use Domain\Files\Actions\DownloadFileAction;
@@ -18,11 +19,19 @@ class GetFileController extends Controller
 
     public function __invoke(
         string $filename,
-    ): BinaryFileResponse {
+    ): Response|BinaryFileResponse {
         $file = UserFile::withTrashed()
             ->where('basename', $filename)
             ->firstOrFail();
 
+        // Check if user can download file
+        if (! $file->owner->canDownload()) {
+            return response([
+                'type'    => 'error',
+                'message' => 'This user action is not allowed.',
+            ], 401);
+        }
+
         if (! Gate::any(['can-edit', 'can-view'], [$file, null])) {
             abort(403, 'Access Denied');
         }
diff --git a/src/Domain/Files/Controllers/FileAccess/VisitorGetFileController.php b/src/Domain/Files/Controllers/FileAccess/VisitorGetFileController.php
index ce143b3c..f236d817 100644
--- a/src/Domain/Files/Controllers/FileAccess/VisitorGetFileController.php
+++ b/src/Domain/Files/Controllers/FileAccess/VisitorGetFileController.php
@@ -2,6 +2,7 @@
 namespace Domain\Files\Controllers\FileAccess;
 
 use Domain\Files\Models\File;
+use Illuminate\Http\Response;
 use Domain\Sharing\Models\Share;
 use App\Http\Controllers\Controller;
 use Domain\Files\Actions\DownloadFileAction;
@@ -26,7 +27,15 @@ class VisitorGetFileController extends Controller
     public function __invoke(
         $filename,
         Share $shared,
-    ): BinaryFileResponse {
+    ): BinaryFileResponse|Response {
+        // Check if user can download file
+        if (! $shared->user->canDownload()) {
+            return response([
+                'type'    => 'error',
+                'message' => 'This user action is not allowed.',
+            ], 401);
+        }
+
         // Check ability to access protected share files
         ($this->protectShareRecord)($shared);
 
diff --git a/src/Support/helpers.php b/src/Support/helpers.php
index 0e3635a7..098fd002 100644
--- a/src/Support/helpers.php
+++ b/src/Support/helpers.php
@@ -33,11 +33,11 @@ if (! function_exists('obfuscate_email')) {
     }
 }
 
-if (! function_exists('get_limitation_driver')) {
+if (! function_exists('get_restriction_driver')) {
     /**
      * Get driver for limitation API
      */
-    function get_limitation_driver(): string
+    function get_restriction_driver(): string
     {
         return match (get_settings('subscription_type')) {
             'fixed'   => 'fixed',
diff --git a/tests/App/Limitations/MeteredBillingLimitationTest.php b/tests/App/Limitations/MeteredBillingLimitationTest.php
deleted file mode 100644
index 3d1d1052..00000000
--- a/tests/App/Limitations/MeteredBillingLimitationTest.php
+++ /dev/null
@@ -1,112 +0,0 @@
-<?php
-namespace Tests\App\Limitations;
-
-use Tests\TestCase;
-use App\Users\Models\User;
-use Illuminate\Support\Facades\DB;
-
-class MeteredBillingLimitationTest extends TestCase
-{
-    public function setUp(): void
-    {
-        parent::setUp();
-
-        DB::table('settings')->insert([
-            'name'  => 'subscription_type',
-            'value' => 'metered',
-        ]);
-    }
-
-    /**
-     * @test
-     */
-    public function it_can_upload()
-    {
-        $user = User::factory()
-            ->hasFailedpayments(2)
-            ->create();
-
-        $this->assertEquals(true, $user->canUpload());
-    }
-
-    /**
-     * @test
-     */
-    public function it_cant_upload_because_user_has_3_failed_payments()
-    {
-        $user = User::factory()
-            ->hasFailedpayments(3)
-            ->create();
-
-        $this->assertEquals(false, $user->canUpload());
-    }
-
-    /**
-     * @test
-     */
-    public function it_can_create_new_folder()
-    {
-        $user = User::factory()
-            ->create();
-
-        // Create basic folder
-        $this
-            ->actingAs($user)
-            ->postJson('/api/create-folder', [
-                'name' => 'New Folder',
-            ])
-            ->assertStatus(201);
-
-        // Create team folder
-        $this
-            ->actingAs($user)
-            ->postJson('/api/teams/folders', [
-                'name'        => 'New Team Folder',
-                'invitations' => [
-                    [
-                        'email'      => 'john@doe.com',
-                        'permission' => 'can-edit',
-                        'type'       => 'invitation',
-                    ],
-                ],
-            ])
-            ->assertStatus(201);
-
-        $this->assertDatabaseCount('folders', 2);
-    }
-
-    /**
-     * @test
-     */
-    public function it_cant_create_new_folder_because_user_has_3_failed_payments()
-    {
-        $user = User::factory()
-            ->hasFailedpayments(3)
-            ->create();
-
-        // Create basic folder
-        $this
-            ->actingAs($user)
-            ->postJson('/api/create-folder', [
-                'name' => 'New Folder',
-            ])
-            ->assertStatus(401);
-
-        // Create team folder
-        $this
-            ->actingAs($user)
-            ->postJson('/api/teams/folders', [
-                'name'        => 'New Folder',
-                'invitations' => [
-                    [
-                        'email'      => 'john@doe.com',
-                        'permission' => 'can-edit',
-                        'type'       => 'invitation',
-                    ],
-                ],
-            ])
-            ->assertStatus(401);
-
-        $this->assertDatabaseCount('folders', 0);
-    }
-}
diff --git a/tests/App/Limitations/DefaultLimitationTest.php b/tests/App/Restrictions/DefaultRestrictionsTest.php
similarity index 75%
rename from tests/App/Limitations/DefaultLimitationTest.php
rename to tests/App/Restrictions/DefaultRestrictionsTest.php
index ae4946b1..3378a612 100644
--- a/tests/App/Limitations/DefaultLimitationTest.php
+++ b/tests/App/Restrictions/DefaultRestrictionsTest.php
@@ -1,13 +1,14 @@
 <?php
-namespace Tests\App\Limitations;
+namespace Tests\App\Restrictions;
 
 use Tests\TestCase;
 use App\Users\Models\User;
 use Domain\Files\Models\File;
+use Domain\Sharing\Models\Share;
 use Domain\Settings\Models\Setting;
 use Domain\Teams\Models\TeamFolderMember;
 
-class DefaultLimitationTest extends TestCase
+class DefaultRestrictionsTest extends TestCase
 {
     /**
      * @test
@@ -130,10 +131,10 @@ class DefaultLimitationTest extends TestCase
             ->create()
             ->each(
                 fn ($member) => TeamFolderMember::factory()
-            ->create([
-                'parent_id' => $user->folders[0]->id,
-                'user_id'   => $member->id,
-            ])
+                    ->create([
+                        'parent_id' => $user->folders[0]->id,
+                        'user_id'   => $member->id,
+                    ])
             );
 
         // Try invite new members, it has to fail
@@ -184,4 +185,54 @@ class DefaultLimitationTest extends TestCase
             ])
             ->assertCreated();
     }
+
+    /**
+     * @test
+     */
+    public function it_can_get_private_file()
+    {
+        $user = User::factory()
+            ->create();
+
+        $file = File::factory()
+            ->create([
+                'user_id'  => $user->id,
+                'basename' => 'fake-file.pdf',
+                'name'     => 'fake-file.pdf',
+            ]);
+
+        // 404 but, ok, because there is not stored temporary file in test
+        $this
+            ->actingAs($user)
+            ->get("file/$file->name")
+            ->assertStatus(404);
+    }
+
+    /**
+     * @test
+     */
+    public function it_can_get_shared_file()
+    {
+        $user = User::factory()
+            ->create();
+
+        $file = File::factory()
+            ->create([
+                'basename' => 'fake-file.pdf',
+                'name'     => 'fake-file.pdf',
+            ]);
+
+        $share = Share::factory()
+            ->create([
+                'item_id'      => $file->id,
+                'user_id'      => $user->id,
+                'type'         => 'file',
+                'is_protected' => false,
+            ]);
+
+        // 404 but, ok, because there is not stored temporary file in test
+        $this
+            ->get("file/$file->name/$share->token")
+            ->assertStatus(404);
+    }
 }
diff --git a/tests/App/Limitations/FixedBillingLimitationTest.php b/tests/App/Restrictions/FixedBillingRestrictionsTest.php
similarity index 70%
rename from tests/App/Limitations/FixedBillingLimitationTest.php
rename to tests/App/Restrictions/FixedBillingRestrictionsTest.php
index ee1b7022..348adefe 100644
--- a/tests/App/Limitations/FixedBillingLimitationTest.php
+++ b/tests/App/Restrictions/FixedBillingRestrictionsTest.php
@@ -1,13 +1,14 @@
 <?php
-namespace Tests\App\Limitations;
+namespace Tests\App\Restrictions;
 
 use Tests\TestCase;
 use App\Users\Models\User;
 use Domain\Files\Models\File;
+use Domain\Sharing\Models\Share;
 use Domain\Settings\Models\Setting;
 use Domain\Teams\Models\TeamFolderMember;
 
-class FixedBillingLimitationTest extends TestCase
+class FixedBillingRestrictionsTest extends TestCase
 {
     public function setUp(): void
     {
@@ -91,10 +92,10 @@ class FixedBillingLimitationTest extends TestCase
             ->create()
             ->each(
                 fn ($member) => TeamFolderMember::factory()
-            ->create([
-                'parent_id' => $user->folders[0]->id,
-                'user_id'   => $member->id,
-            ])
+                    ->create([
+                        'parent_id' => $user->folders[0]->id,
+                        'user_id'   => $member->id,
+                    ])
             );
 
         // Try invite new members, it has to fail
@@ -145,4 +146,54 @@ class FixedBillingLimitationTest extends TestCase
             ])
             ->assertCreated();
     }
+
+    /**
+     * @test
+     */
+    public function it_can_get_private_file()
+    {
+        $user = User::factory()
+            ->create();
+
+        $file = File::factory()
+            ->create([
+                'user_id'  => $user->id,
+                'basename' => 'fake-file.pdf',
+                'name'     => 'fake-file.pdf',
+            ]);
+
+        // 404 but, ok, because there is not stored temporary file in test
+        $this
+            ->actingAs($user)
+            ->get("file/$file->name")
+            ->assertStatus(404);
+    }
+
+    /**
+     * @test
+     */
+    public function it_can_get_shared_file()
+    {
+        $user = User::factory()
+            ->create();
+
+        $file = File::factory()
+            ->create([
+                'basename' => 'fake-file.pdf',
+                'name'     => 'fake-file.pdf',
+            ]);
+
+        $share = Share::factory()
+            ->create([
+                'item_id'      => $file->id,
+                'user_id'      => $user->id,
+                'type'         => 'file',
+                'is_protected' => false,
+            ]);
+
+        // 404 but, ok, because there is not stored temporary file in test
+        $this
+            ->get("file/$file->name/$share->token")
+            ->assertStatus(404);
+    }
 }
diff --git a/tests/App/Restrictions/MeteredBillingRestrictionsTest.php b/tests/App/Restrictions/MeteredBillingRestrictionsTest.php
new file mode 100644
index 00000000..caf65836
--- /dev/null
+++ b/tests/App/Restrictions/MeteredBillingRestrictionsTest.php
@@ -0,0 +1,216 @@
+<?php
+namespace Tests\App\Restrictions;
+
+use Tests\TestCase;
+use App\Users\Models\User;
+use Domain\Files\Models\File;
+use Domain\Sharing\Models\Share;
+use Illuminate\Support\Facades\DB;
+
+class MeteredBillingRestrictionsTest extends TestCase
+{
+    public function setUp(): void
+    {
+        parent::setUp();
+
+        DB::table('settings')->insert([
+            'name'  => 'subscription_type',
+            'value' => 'metered',
+        ]);
+    }
+
+    /**
+     * @test
+     */
+    public function it_can_upload()
+    {
+        $user = User::factory()
+            ->hasFailedpayments(2)
+            ->create();
+
+        $this->assertEquals(true, $user->canUpload());
+    }
+
+    /**
+     * @test
+     */
+    public function it_cant_upload_because_user_has_3_failed_payments()
+    {
+        $user = User::factory()
+            ->hasFailedpayments(3)
+            ->create();
+
+        $this->assertEquals(false, $user->canUpload());
+    }
+
+    /**
+     * @test
+     */
+    public function it_can_create_new_folder()
+    {
+        $user = User::factory()
+            ->create();
+
+        // Create basic folder
+        $this
+            ->actingAs($user)
+            ->postJson('/api/create-folder', [
+                'name' => 'New Folder',
+            ])
+            ->assertStatus(201);
+
+        // Create team folder
+        $this
+            ->actingAs($user)
+            ->postJson('/api/teams/folders', [
+                'name'        => 'New Team Folder',
+                'invitations' => [
+                    [
+                        'email'      => 'john@doe.com',
+                        'permission' => 'can-edit',
+                        'type'       => 'invitation',
+                    ],
+                ],
+            ])
+            ->assertStatus(201);
+
+        $this->assertDatabaseCount('folders', 2);
+    }
+
+    /**
+     * @test
+     */
+    public function it_cant_create_new_folder_because_user_has_3_failed_payments()
+    {
+        $user = User::factory()
+            ->hasFailedpayments(3)
+            ->create();
+
+        // Create basic folder
+        $this
+            ->actingAs($user)
+            ->postJson('/api/create-folder', [
+                'name' => 'New Folder',
+            ])
+            ->assertStatus(401);
+
+        // Create team folder
+        $this
+            ->actingAs($user)
+            ->postJson('/api/teams/folders', [
+                'name'        => 'New Folder',
+                'invitations' => [
+                    [
+                        'email'      => 'john@doe.com',
+                        'permission' => 'can-edit',
+                        'type'       => 'invitation',
+                    ],
+                ],
+            ])
+            ->assertStatus(401);
+
+        $this->assertDatabaseCount('folders', 0);
+    }
+
+    /**
+     * @test
+     */
+    public function it_cant_get_private_file_because_user_has_3_failed_payments()
+    {
+        $user = User::factory()
+            ->hasFailedpayments(3)
+            ->create();
+
+        $file = File::factory()
+            ->create([
+                'user_id'  => $user->id,
+                'basename' => 'fake-file.pdf',
+                'name'     => 'fake-file.pdf',
+            ]);
+
+        $this
+            ->actingAs($user)
+            ->get("file/$file->name")
+            ->assertStatus(401);
+    }
+
+    /**
+     * @test
+     */
+    public function it_can_get_private_file()
+    {
+        $user = User::factory()
+            ->create();
+
+        $file = File::factory()
+            ->create([
+                'user_id'  => $user->id,
+                'basename' => 'fake-file.pdf',
+                'name'     => 'fake-file.pdf',
+            ]);
+
+        // 404 but, ok, because there is not stored temporary file in test
+        $this
+            ->actingAs($user)
+            ->get("file/$file->name")
+            ->assertStatus(404);
+    }
+
+    /**
+     * @test
+     */
+    public function it_cant_get_shared_file_because_user_has_3_failed_payments()
+    {
+        $user = User::factory()
+            ->hasFailedpayments(3)
+            ->create();
+
+        $file = File::factory()
+            ->create([
+                'user_id'  => $user->id,
+                'basename' => 'fake-file.pdf',
+                'name'     => 'fake-file.pdf',
+            ]);
+
+        $share = Share::factory()
+            ->create([
+                'item_id'      => $file->id,
+                'user_id'      => $user->id,
+                'type'         => 'file',
+                'is_protected' => false,
+            ]);
+
+        $this
+            ->get("file/$file->name/$share->token")
+            ->assertStatus(401);
+    }
+
+    /**
+     * @test
+     */
+    public function it_can_get_shared_file()
+    {
+        $user = User::factory()
+            ->create();
+
+        $file = File::factory()
+            ->create([
+                'user_id'  => $user->id,
+                'basename' => 'fake-file.pdf',
+                'name'     => 'fake-file.pdf',
+            ]);
+
+        $share = Share::factory()
+            ->create([
+                'item_id'      => $file->id,
+                'user_id'      => $user->id,
+                'type'         => 'file',
+                'is_protected' => false,
+            ]);
+
+        // 404 but, ok, because there is not stored temporary file in test
+        $this
+            ->get("file/$file->name/$share->token")
+            ->assertStatus(404);
+    }
+}
diff --git a/tests/App/Limitations/LimitationTest.php b/tests/App/Restrictions/RestrictionsTest.php
similarity index 72%
rename from tests/App/Limitations/LimitationTest.php
rename to tests/App/Restrictions/RestrictionsTest.php
index d80753b0..1c7254df 100644
--- a/tests/App/Limitations/LimitationTest.php
+++ b/tests/App/Restrictions/RestrictionsTest.php
@@ -1,10 +1,10 @@
 <?php
-namespace Tests\App\Limitations;
+namespace Tests\App\Restrictions;
 
 use Tests\TestCase;
 use Domain\Settings\Models\Setting;
 
-class LimitationTest extends TestCase
+class RestrictionsTest extends TestCase
 {
     /**
      * @test
@@ -17,7 +17,7 @@ class LimitationTest extends TestCase
             'value' => 'metered',
         ]);
 
-        $this->assertEquals('metered', get_limitation_driver());
+        $this->assertEquals('metered', get_restriction_driver());
     }
     /**
      * @test
@@ -30,7 +30,7 @@ class LimitationTest extends TestCase
             'value' => 'fixed',
         ]);
 
-        $this->assertEquals('fixed', get_limitation_driver());
+        $this->assertEquals('fixed', get_restriction_driver());
     }
     /**
      * @test
@@ -42,6 +42,6 @@ class LimitationTest extends TestCase
 
         $subscriptionType?->delete();
 
-        $this->assertEquals('default', get_limitation_driver());
+        $this->assertEquals('default', get_restriction_driver());
     }
 }