updated tests to reflect updates