Replace a couple of lines that were accidentally removed.