fix method not returning $this