Fixed import in migration 0001