Updated to be compatible with StateBasedGame inherit change