Get the 2 tests to pass