fixing minor bug in schema