diff --git a/GameEngine/src/main/java/gameEngine/GameEditor.java b/GameEngine/src/main/java/gameEngine/GameEditor.java
index 1d4b30e..1e6662e 100644
--- a/GameEngine/src/main/java/gameEngine/GameEditor.java
+++ b/GameEngine/src/main/java/gameEngine/GameEditor.java
@@ -1,5 +1,6 @@
 package gameEngine;
 
+import gameEngine.scenes.EditorScene;
 import gameEngine.views.Button;
 import gameEngine.scenes.Scene;
 import gameEngine.views.Sprite;
@@ -7,7 +8,7 @@
 import gameEngine.views.Window;
 
 public class GameEditor {
-    private Scene scene;
+    private EditorScene scene;
     private final String frameImagePath = "GameEngine/resources/EditorFrame.png";
 
     private Sprite[] EditorFrameSprite = new Sprite[4];
@@ -22,7 +23,7 @@
     private Text createEntityViewButtonText;
 
     public GameEditor(Scene scene) {
-        this.scene = scene;
+        this.scene = (EditorScene) scene;
         initializeEditorComponents();
     }
 
@@ -53,7 +54,7 @@
     }
 
     public void setScene(Scene newScene) {
-        this.scene = newScene;
+        this.scene = (EditorScene) newScene;
         setButtonListeners();
     }