updated tests to pass