Fixed import error in migration (#1607)