updated tests to fix incorrect expectations