Minor tweak with returning self.