Fix incorrect import for factory