Add missing return statement $this