Add setSchema to test