Revert accidentally removed lines