Add return $this to the set() method of the Model class