Fix 'logprobs' output