updates tests to reflect changes