diff --git a/w2v-server.pl b/w2v-server.pl
index 35b0295..4aff19d 100755
--- a/w2v-server.pl
+++ b/w2v-server.pl
@@ -849,8 +849,8 @@
 									word_activation_sum += syn_nbs[a]->best[b].activation;
                   if(syn_nbs[a]->best[b].activation > best[i].max_activation)
 										best[i].max_activation = syn_nbs[a]->best[b].activation;
-                  if(syn_nbs[a]->best[b].cprobability > best[i].heat[wpos])
-										best[i].heat[wpos] = syn_nbs[a]->best[b].cprobability;
+                  if(best[i].heat[wpos] == 0 )
+										best[i].heat[wpos] = syn_nbs[a]->best[b].activation;
 								}
 						}
 					}