Fix `self.main` reference