Use merge for dict in expand_schema