Fix a small mistake. Add a line back.