fix schema to align with data generated