updating tests to reflect changes.