Fixed episode 9 bug