Fixed tests by changing the expected output to match the changed result output.