Updated schema to fix minor bug