fixed failing tests, oops