Fixed a left-over of a reversed refactor