Add correct schema