diff --git a/app/handlers/subscription.py b/app/handlers/subscription.py
index 55cde773..7a2dbf71 100644
--- a/app/handlers/subscription.py
+++ b/app/handlers/subscription.py
@@ -21,7 +21,7 @@ from app.database.crud.user import subtract_user_balance
 from app.database.crud.transaction import create_transaction, get_user_transactions
 from app.database.models import (
     User, TransactionType, SubscriptionStatus, 
-    SubscriptionServer  
+    SubscriptionServer, Subscription 
 )
 from app.keyboards.inline import (
     get_subscription_keyboard, get_trial_keyboard,