Add missing CoreData import