Use the correct schema when initializing the model