Fix test by restoring line that was accidentally removed.