Undo accidentally removed if statement