Must change UIMap class checks to UIGameMap.