Add back accidentally removed line