diff --git a/tests/openwisp2/sample_users/migrations/0002_default_groups_and_permissions.py b/tests/openwisp2/sample_users/migrations/0002_default_groups_and_permissions.py
index 227b108..3d0597b 100644
--- a/tests/openwisp2/sample_users/migrations/0002_default_groups_and_permissions.py
+++ b/tests/openwisp2/sample_users/migrations/0002_default_groups_and_permissions.py
@@ -3,6 +3,7 @@
 import swapper
 from django.db import migrations
 from openwisp_users.migrations import (
+    add_api_key_permissions_to_admins,
     allow_admins_change_organization,
     create_default_groups,
     set_default_organization_uuid,
@@ -30,4 +31,7 @@ class Migration(migrations.Migration):
         migrations.RunPython(
             allow_admins_change_organization, reverse_code=migrations.RunPython.noop
         ),
+        migrations.RunPython(
+            add_api_key_permissions_to_admins, reverse_code=migrations.RunPython.noop
+        ),
     ]
diff --git a/tests/openwisp2/sample_users/migrations/0003_user_expiration_date_user_user_active_expiry_idx.py b/tests/openwisp2/sample_users/migrations/0003_user_expiration_date_user_user_active_expiry_idx.py
index 7f02772..a51f5ad 100644
--- a/tests/openwisp2/sample_users/migrations/0003_user_expiration_date_user_user_active_expiry_idx.py
+++ b/tests/openwisp2/sample_users/migrations/0003_user_expiration_date_user_user_active_expiry_idx.py
@@ -7,6 +7,7 @@ class Migration(migrations.Migration):
 
     dependencies = [
         ("auth", "0012_alter_user_first_name_max_length"),
+        ("authtoken", "0001_initial"),
         ("sample_users", "0002_default_groups_and_permissions"),
     ]
 
@@ -21,6 +22,18 @@ class Migration(migrations.Migration):
                 verbose_name="expiration date",
             ),
         ),
+        migrations.CreateModel(
+            name="ApiKey",
+            fields=[],
+            options={
+                "verbose_name": "API key",
+                "verbose_name_plural": "API keys",
+                "proxy": True,
+                "indexes": [],
+                "constraints": [],
+            },
+            bases=("authtoken.token",),
+        ),
         migrations.AddIndex(
             model_name="user",
             index=models.Index(
diff --git a/tests/openwisp2/sample_users/models.py b/tests/openwisp2/sample_users/models.py
index 35ac324..5f1d721 100644
--- a/tests/openwisp2/sample_users/models.py
+++ b/tests/openwisp2/sample_users/models.py
@@ -47,3 +47,6 @@ class OrganizationInvitation(AbstractOrganizationInvitation):
 
 class Group(BaseGroup, AbstractGroup):
     pass
+
+
+from openwisp_users.base.models import ApiKey  # noqa