Restore accidental deletion of code