This comment calls out contexts but this is looking for `klass` objects not necessarily DataContext objects. Remove or reword this comment.