Use larger model for unit tests