diff --git a/src/main/java/cactusServer/models/StageManager.java b/src/main/java/cactusServer/models/StageManager.java
index 75987e8..4a65953 100644
--- a/src/main/java/cactusServer/models/StageManager.java
+++ b/src/main/java/cactusServer/models/StageManager.java
@@ -1,5 +1,6 @@
 package cactusServer.models;
 
+import java.util.HashMap;
 import java.util.HashSet;
 
 import javax.inject.Singleton;
@@ -9,7 +10,7 @@
 @Singleton
 public class StageManager {
 	private static StageManager theInstance = null;
-	private HashSet<Model3D> model = new HashSet<>();
+	private HashMap<Integer,Model3D> model = new HashMap<>();
 	private StageManager() {
 
 	}
@@ -22,6 +23,6 @@
 	}
 
 	public Model3D getStage(int stageID) {
-		return null;
+		return model.get(stageID);
 	}
 }