add correct schema generation