diff --git a/src/components/GenerateCourse/AICourseContent.tsx b/src/components/GenerateCourse/AICourseContent.tsx
index 8405c2e8e..d182be30a 100644
--- a/src/components/GenerateCourse/AICourseContent.tsx
+++ b/src/components/GenerateCourse/AICourseContent.tsx
@@ -18,7 +18,7 @@ import { CheckIcon } from '../ReactIcons/CheckIcon';
 import { ErrorIcon } from '../ReactIcons/ErrorIcon';
 import { AICourseLimit } from './AICourseLimit';
 import { AICourseSidebarModuleList } from './AICourseSidebarModuleList';
-import { AICourseModuleView } from './AICourseModuleView';
+import { AICourseLesson } from './AICourseLesson';
 import { UpgradeAccountModal } from '../Billing/UpgradeAccountModal';
 import { AILimitsPopup } from './AILimitsPopup';
 import { RegenerateOutline } from './RegenerateOutline';
@@ -387,7 +387,7 @@ export function AICourseContent(props: AICourseContentProps) {
           )}
         >
           {viewMode === 'module' && (
-            <AICourseModuleView
+            <AICourseLesson
               courseSlug={courseSlug!}
               activeModuleIndex={activeModuleIndex}
               totalModules={totalModules}
diff --git a/src/components/GenerateCourse/AICourseModuleView.tsx b/src/components/GenerateCourse/AICourseLesson.tsx
similarity index 96%
rename from src/components/GenerateCourse/AICourseModuleView.tsx
rename to src/components/GenerateCourse/AICourseLesson.tsx
index 975ffa293..8b4418261 100644
--- a/src/components/GenerateCourse/AICourseModuleView.tsx
+++ b/src/components/GenerateCourse/AICourseLesson.tsx
@@ -27,7 +27,7 @@ import { AICourseFollowUp } from './AICourseFollowUp';
 import './AICourseFollowUp.css';
 import { useIsPaidUser } from '../../queries/billing';
 
-type AICourseModuleViewProps = {
+type AICourseLessonProps = {
   courseSlug: string;
 
   activeModuleIndex: number;
@@ -43,7 +43,7 @@ type AICourseModuleViewProps = {
   onUpgrade: () => void;
 };
 
-export function AICourseModuleView(props: AICourseModuleViewProps) {
+export function AICourseLesson(props: AICourseLessonProps) {
   const {
     courseSlug,
 
@@ -222,11 +222,11 @@ export function AICourseModuleView(props: AICourseModuleViewProps) {
           </div>
 
           {!isGenerating && !isLoading && (
-            <>
+            <div className="absolute right-3 top-3 flex items-center justify-between gap-2">
               <button
                 disabled={isLoading || isTogglingDone}
                 className={cn(
-                  'absolute right-3 top-3 flex items-center gap-1.5 rounded-full bg-black py-1 pl-2 pr-3 text-sm text-white hover:bg-gray-800 disabled:opacity-50 max-lg:text-xs',
+                  'flex items-center gap-1.5 rounded-full bg-black py-1 pl-2 pr-3 text-sm text-white hover:bg-gray-800 disabled:opacity-50 max-lg:text-xs',
                   isLessonDone
                     ? 'bg-red-500 hover:bg-red-600'
                     : 'bg-green-500 hover:bg-green-600',
@@ -258,7 +258,7 @@ export function AICourseModuleView(props: AICourseModuleViewProps) {
                   </>
                 )}
               </button>
-            </>
+            </div>
           )}
         </div>
 
diff --git a/src/components/GenerateCourse/AICourseSidebarModuleList.tsx b/src/components/GenerateCourse/AICourseSidebarModuleList.tsx
index 20b0720d6..7bbf557ee 100644
--- a/src/components/GenerateCourse/AICourseSidebarModuleList.tsx
+++ b/src/components/GenerateCourse/AICourseSidebarModuleList.tsx
@@ -19,8 +19,8 @@ type AICourseModuleListProps = {
 
   setSidebarOpen: (open: boolean) => void;
 
-  viewMode: 'module' | 'full';
-  setViewMode: (mode: 'module' | 'full') => void;
+  viewMode: 'module' | 'outline';
+  setViewMode: (mode: 'module' | 'outline') => void;
 
   expandedModules: Record<number, boolean>;
   setExpandedModules: Dispatch<SetStateAction<Record<number, boolean>>>;