Fix a little code.