diff --git a/Frontend/lib/mih_packages/mzansi_ai/ai_chat.dart b/Frontend/lib/mih_packages/mzansi_ai/ai_chat.dart
index 73834709..6a67b745 100644
--- a/Frontend/lib/mih_packages/mzansi_ai/ai_chat.dart
+++ b/Frontend/lib/mih_packages/mzansi_ai/ai_chat.dart
@@ -166,6 +166,14 @@ class _AiChatState extends State<AiChat> {
     );
   }
 
+  @override
+  void dispose() {
+    // TODO: implement dispose
+    super.dispose();
+    _modelCopntroller.dispose();
+    client.endSession();
+  }
+
   @override
   void initState() {
     super.initState();