diff --git a/src/Welsh Vocabulary Tutor/src/main/java/uk/ac/aber/cs22120/group20/javafx/FlashcardController.java b/src/Welsh Vocabulary Tutor/src/main/java/uk/ac/aber/cs22120/group20/javafx/FlashcardController.java
index 61a1921..cc9d12d 100644
--- a/src/Welsh Vocabulary Tutor/src/main/java/uk/ac/aber/cs22120/group20/javafx/FlashcardController.java	
+++ b/src/Welsh Vocabulary Tutor/src/main/java/uk/ac/aber/cs22120/group20/javafx/FlashcardController.java	
@@ -10,6 +10,7 @@ import javafx.scene.text.Text;
 import javafx.scene.transform.Rotate;
 import javafx.util.Duration;
 import uk.ac.aber.cs22120.group20.javafx.Application;
+import uk.ac.aber.cs22120.group20.selfassessment.AssessmentGenerator;
 
 import java.io.IOException;
 
@@ -108,7 +109,7 @@ public class FlashcardController {
 
    @FXML
    private void switchToAddWord() throws IOException {
-      ScreenSwitch.swap(ScreenSwitch.SceneEnum.addWordScene);
+      AssessmentGenerator.generateAssessment(Application.dictionary);
    }
 
 }