From 65ecdac160efb6686835bc23e71b484d83a1112e Mon Sep 17 00:00:00 2001
From: gosticks <wlad.meixner@protonmail.com>
Date: Mon, 22 Jan 2018 16:31:59 +0100
Subject: [PATCH] fixed balance calculation

---
 src/main/java/iotaFlashWrapper/Main.java | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/src/main/java/iotaFlashWrapper/Main.java b/src/main/java/iotaFlashWrapper/Main.java
index 1a655eb..b881892 100644
--- a/src/main/java/iotaFlashWrapper/Main.java
+++ b/src/main/java/iotaFlashWrapper/Main.java
@@ -273,15 +273,11 @@ public class Main {
     }
 
     public static double getBalanceOfUser(UserObject user) {
-        double balance = 0.0;
+        double balance = user.getFlash().getDeposits().get(user.getUserIndex());
         Map<String, Integer> transfers = user.getFlash().getOutputs();
-        if (transfers.size() == 0) {
-            return 0.0;
-        }
-
         for (Map.Entry<String, Integer> transfer : transfers.entrySet()) {
             if (transfer.getKey().equals(user.getAddress())) {
-                    balance += transfer.getValue() / 2;
+                    balance += transfer.getValue();
                 }
         }