fix the 'hidden probs update error' issue