categoryProbability command fix