Is there a way to make it so that this function only needs to take the `course_key` rather than the `course`? That way we'd avoid having to fetch the course for courses not using this feature (would still need to fetch the course around L23 for courses with the feature).