Ensure shadow is using the right model