diff --git a/src/components/GenerateCourse/AICourseLesson.tsx b/src/components/GenerateCourse/AICourseLesson.tsx
index a1aa1e746..2b0312c6f 100644
--- a/src/components/GenerateCourse/AICourseLesson.tsx
+++ b/src/components/GenerateCourse/AICourseLesson.tsx
@@ -27,6 +27,7 @@ import { queryClient } from '../../stores/query-client';
import { AICourseFollowUp } from './AICourseFollowUp';
import './AICourseFollowUp.css';
import { RegenerateLesson } from './RegenerateLesson';
+import { TestMyKnowledgeAction } from './TestMyKnowledgeAction';
type AICourseLessonProps = {
courseSlug: string;
@@ -203,7 +204,9 @@ export function AICourseLesson(props: AICourseLessonProps) {
isLoading;
const cantGoBack =
- (activeModuleIndex === 0 && activeLessonIndex === 0) || isGenerating || isLoading;
+ (activeModuleIndex === 0 && activeLessonIndex === 0) ||
+ isGenerating ||
+ isLoading;
return (
@@ -319,6 +322,14 @@ export function AICourseLesson(props: AICourseLessonProps) {
)}
+ {!isLoading && !isGenerating && !error && (
+
+ )}
+