self.agent -> self.context