fix return $this