edit tests to pass