Fix mistake in probabilties update.