Fix imports for schema/migrations