Defined an explicit schema
Fixed an issue with the schema