diff --git a/homelab-ai-bot/telegram_bot.py b/homelab-ai-bot/telegram_bot.py
index 7adfc37c..735d5420 100644
--- a/homelab-ai-bot/telegram_bot.py
+++ b/homelab-ai-bot/telegram_bot.py
@@ -59,9 +59,13 @@ def _release_lock():
         except OSError:
             pass
 
-from telegram import BotCommand, Update, ReplyKeyboardMarkup, KeyboardButton
+from telegram import (
+    BotCommand, Update, ReplyKeyboardMarkup, KeyboardButton,
+    InlineKeyboardButton, InlineKeyboardMarkup,
+)
 from telegram.ext import (
-    Application, CommandHandler, MessageHandler, filters, ContextTypes,
+    Application, CommandHandler, MessageHandler, CallbackQueryHandler,
+    filters, ContextTypes,
 )
 
 BOT_COMMANDS = [