Modified tests to ensure setters returned $this