Use appropriate model types