Fix a little important code bug.