Fix after code review

https://github.com/jmmastey/level_up_exercises/pull/312