diff --git a/handlers/keys/key_management.py b/handlers/keys/key_management.py
index 70701242..fb0f825b 100644
--- a/handlers/keys/key_management.py
+++ b/handlers/keys/key_management.py
@@ -90,6 +90,10 @@ async def handle_key_creation(
     if trial_status == 0:
         expiry_time = current_time + timedelta(days=TRIAL_TIME)
         logger.info(f"Assigned 1-day trial to user {tg_id}.")
+        
+        await session.execute(
+            "UPDATE connections SET trial = 1 WHERE tg_id = $1", tg_id
+        )
         await create_key(tg_id, expiry_time, state, session, message_or_query)
     else:
         builder = InlineKeyboardBuilder()
@@ -118,6 +122,7 @@ async def handle_key_creation(
         await state.set_state(Form.waiting_for_server_selection)
 
 
+
 @router.callback_query(F.data.startswith("select_plan_"))
 async def select_tariff_plan(callback_query: CallbackQuery, state: FSMContext, session: Any):
     tg_id = callback_query.message.chat.id