updates tests to reflect feature