Correct a minor bug