fixed unit tests expected output