Fix incorrect import