Fixed regression from last commit :P