Change to `self.assertEqual`