[Migration] add missing import