Probabilities are not altered for non-uniform datasets

In this case, all new entries are given 0 probability

com.unity.perception/Editor/Randomization/VisualElements/Parameter/ParameterElement.cs

             listView.style.flexGrow = 1.0f;
             listView.style.height = new StyleLength(listView.itemHeight * 4);
 
+            var uniformToggle = template.Q<Toggle>("uniform");
+
             VisualElement MakeItem()
             {
                 return new CategoricalOptionElement(
                     {
                         optionsProperty.DeleteArrayElementAtIndex(i);
                     }
-
-                    ResetProbabilities();
 
                     m_SerializedProperty.serializedObject.ApplyModifiedProperties();
 
                         break;
                 }
 
-                ResetProbabilities();
+                // New items probability will be 0, unless uniform toggle is true
+                probabilitiesProperty.GetArrayElementAtIndex(probabilitiesProperty.arraySize - 1).floatValue = 0;
+
+                if (uniformToggle.value)
+                    ResetProbabilities();
 
                 m_SerializedProperty.serializedObject.ApplyModifiedProperties();
                 listView.itemsSource = categoricalParameter.probabilities;
                     {
                         var optionProperty = optionsProperty.GetArrayElementAtIndex(optionsIndex + i);
                         optionProperty.objectReferenceValue = categories[i];
+                        probabilitiesProperty.GetArrayElementAtIndex(i).floatValue = 0;
-                    ResetProbabilities();
+                    if (uniformToggle.value)
+                        ResetProbabilities();
 
                     m_SerializedProperty.serializedObject.ApplyModifiedProperties();
                     listView.itemsSource = categoricalParameter.probabilities;
                     evt.StopImmediatePropagation();
             });
 
-            var uniformToggle = template.Q<Toggle>("uniform");
+
             var uniformProperty = m_SerializedProperty.FindPropertyRelative("uniform");
             uniformToggle.BindProperty(uniformProperty);