Fixed a small bug with the schema