From adcdc597bbc8d7cd88208e135e5f9665107aa278 Mon Sep 17 00:00:00 2001
From: Youngjoon Lee <5462944+youngjoon-lee@users.noreply.github.com>
Date: Tue, 21 May 2024 15:01:07 +0900
Subject: [PATCH] fix: handling weight properly

---
 mixnet/v2/sim/node.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mixnet/v2/sim/node.py b/mixnet/v2/sim/node.py
index 2d349c2..77e688d 100644
--- a/mixnet/v2/sim/node.py
+++ b/mixnet/v2/sim/node.py
@@ -55,7 +55,7 @@ class Node:
 
     def real_message_prob(self):
         weight = self.config.mixnet.real_message_prob_weights[self.id] \
-            if self.id < len(self.config.mixnet.real_message_prob_weights) else 0
+            if self.id < len(self.config.mixnet.real_message_prob_weights) else 1
         return self.config.mixnet.real_message_prob * weight
 
     def create_message(self, message_type: MessageType) -> SphinxPacket | bytes: