Fixed migration import error