Figured out how to make the tests pass.