fixing schema generation on tests