Changed name on method getHero() to getHeroModel() in class PlayState