Didn't follow the rest of the previous code...