fixed missing import on migration script