diff --git a/Source/ThunderSVMWrapper.cpp b/Source/ThunderSVMWrapper.cpp
index 4d14542..6743ebb 100644
--- a/Source/ThunderSVMWrapper.cpp
+++ b/Source/ThunderSVMWrapper.cpp
@@ -149,6 +149,12 @@ extern "C" {
         instances.clear();
         for (int r = 0; r < rows; r++)
             predictions[r] = (float)predicted[r];
+        if (probabilities != NULL && sw->param.probability) {
+            vector<float> probs = sw->model->get_prob_predict();
+            for (int r = 0; r < rows; r++)
+                for (int c = 0; c < sw->classes; c++)
+                    probabilities[c + r * sw->classes] = probs[c + r * sw->classes];
+        }
         predicted.clear();
 
         return;
diff --git a/Unity/Plugins/ThunderSVMWrapper.dll b/Unity/Plugins/ThunderSVMWrapper.dll
index 14e1211..ff85cdb 100644
Binary files a/Unity/Plugins/ThunderSVMWrapper.dll and b/Unity/Plugins/ThunderSVMWrapper.dll differ