Missing a variable in the refactor XD