Fix bug in simplify unit test