updates tests to pass