Good point. The course is used by the new ```unit_container_xblock_component.html``` template in order to determine the URL for the "View" button. I've moved the logic as you suggest.