Adding missing $this->