Missed something else in the refacter