A little cleanup to help make your tests pass