switch back to prob instead of log prob