Set correct schema