diff --git a/api/kiwi_vpn_api/db/__init__.py b/api/kiwi_vpn_api/db/__init__.py
index 14006e1..5538a57 100644
--- a/api/kiwi_vpn_api/db/__init__.py
+++ b/api/kiwi_vpn_api/db/__init__.py
@@ -3,7 +3,7 @@ Package `db`: ORM and schemas for database content.
 """
 
 from .connection import Connection
-from .device import Device, DeviceBase, DeviceCreate, DeviceRead
+from .device import Device, DeviceBase, DeviceCreate, DeviceRead, DeviceStatus
 from .tag import TagValue
 from .user import User, UserBase, UserCreate, UserRead
 
@@ -13,6 +13,7 @@ __all__ = [
     "DeviceBase",
     "DeviceCreate",
     "DeviceRead",
+    "DeviceStatus",
     "User",
     "UserBase",
     "UserCreate",
diff --git a/api/kiwi_vpn_api/routers/device.py b/api/kiwi_vpn_api/routers/device.py
index 28b0450..57b415b 100644
--- a/api/kiwi_vpn_api/routers/device.py
+++ b/api/kiwi_vpn_api/routers/device.py
@@ -3,9 +3,8 @@
 """
 
 from fastapi import APIRouter, Depends, HTTPException, status
-from kiwi_vpn_api.db.device import DeviceStatus
 
-from ..db import Device, DeviceCreate, DeviceRead, User
+from ..db import Device, DeviceCreate, DeviceRead, DeviceStatus, User
 from ..easyrsa import EASYRSA, DistinguishedName
 from ._common import (Responses, get_current_user, get_device_by_id,
                       get_user_by_name)