I don't see where this gets set. It also doesn't seem to discriminate based on `self.which_set`.