updated tests to reflect the changes from previous commit