1 files changed, 7 insertions, 4 deletions
diff --git a/edit.h b/edit.h
index 854a4b5..460fb89 100644
--- a/edit.h
+++ b/edit.h
@@ -1,7 +1,7 @@
 #ifndef EDIT_H_
 #define EDIT_H_
 
-#include "level.h"
+#include "Level.h"
 #include "geometry.h"
 
 #define EDIT_MODE_VIEW     0
@@ -10,13 +10,16 @@
 
 int getEditMode();
 
-ROOM *getActiveRoom();
-void setActiveRoom(ROOM *room);
+Level *getLevel();
+void setLevel(Level *l);
+
+Room *getActiveRoom();
+void setActiveRoom(Room *room);
 
 void startAddMode();
 void endAddMode();
 
-ROOM *getHoveredRoom();
+Room *getHoveredRoom();
 
 Vertex *getHoveredVertex();
 void setHoveredVertex(Vertex *v);