Add test for returning self.

Small refactor.