diff --git a/src/components/GenerateCourse/AICourseContent.tsx b/src/components/GenerateCourse/AICourseContent.tsx
index 37c197cc1..6b014eb56 100644
--- a/src/components/GenerateCourse/AICourseContent.tsx
+++ b/src/components/GenerateCourse/AICourseContent.tsx
@@ -245,16 +245,18 @@ export function AICourseContent(props: AICourseContentProps) {
               />
             </div>
 
-            <button
-              onClick={() => setIsAIChatsOpen(!isAIChatsOpen)}
-              className="flex items-center justify-center text-gray-400 shadow-sm transition-colors hover:bg-gray-50 hover:text-gray-900 lg:hidden"
-            >
-              {isAIChatsOpen ? (
-                <MessageCircleOffIcon size={17} strokeWidth={3} />
-              ) : (
-                <MessageCircleIcon size={17} strokeWidth={3} />
-              )}
-            </button>
+            {viewMode === 'module' && (
+              <button
+                onClick={() => setIsAIChatsOpen(!isAIChatsOpen)}
+                className="flex items-center justify-center text-gray-400 shadow-sm transition-colors hover:bg-gray-50 hover:text-gray-900 lg:hidden"
+              >
+                {isAIChatsOpen ? (
+                  <MessageCircleOffIcon size={17} strokeWidth={3} />
+                ) : (
+                  <MessageCircleIcon size={17} strokeWidth={3} />
+                )}
+              </button>
+            )}
 
             <button
               onClick={() => setSidebarOpen(!sidebarOpen)}