diff --git a/src/NlpTools/Models/Lda.php b/src/NlpTools/Models/Lda.php
index bec01d2..efde3cf 100644
--- a/src/NlpTools/Models/Lda.php
+++ b/src/NlpTools/Models/Lda.php
@@ -240,7 +240,7 @@ public function getDocumentsPerTopicsProbabilities($limit_docs=-1)
          $count_topics_docs = array();
          foreach ($this->count_docs_topics as $doc=>$topics) {
              foreach ($topics as $t=>$c)
-                $count_topics_docs[$doc][$t]++;
+                $count_topics_docs[$doc][$t]+=$c;
          }
 
          foreach ($p_t_d as $topic=>&$p) {
@@ -253,7 +253,7 @@ public function getDocumentsPerTopicsProbabilities($limit_docs=-1)
              }
          }
 
-         return $p;
+         return $p_t_d;
      }
 
      /**