Fix wrongly deleted line

Add back a line that was accidentally deleted by previous commit