Fix incorrect import.