Correct a compile bug