got 2 tests to pass