diff --git a/groq_chat/bot.py b/groq_chat/bot.py
index 29a32f2..f0054bf 100644
--- a/groq_chat/bot.py
+++ b/groq_chat/bot.py
@@ -60,7 +60,7 @@ def start_bot():
 
     # Add persistence if available
     if persistence:
-        app_builder.persistence = persistence
+        app_builder.persistence(persistence)
 
     # Build the app
     app = app_builder.build()