Add an accidentally removed line. Oops