Fix invalid class import.