Correct a import bug