Minor testing config issue for model wrapper