Correct expected output to match python output (string comparison)