undo small regression from previous commit