Be sure to use `self.assertIn()`! I'm going to merge this but this should be a cleanup task