diff --git a/src/main/java/weka/finito/client.java b/src/main/java/weka/finito/client.java
index 3856e31..3486cf1 100644
--- a/src/main/java/weka/finito/client.java
+++ b/src/main/java/weka/finito/client.java
@@ -417,7 +417,7 @@ public void run() {
 					hashed_classification.put(hash(aClass), aClass);
 				}
 				// Make sure level-sites got everything...
-				Thread.sleep(2000);
+				Thread.sleep(2500);
 			}
 			else {
 				System.out.println("Not contacting server-site. Seems you just want to test on the" +
diff --git a/src/main/java/weka/finito/structs/NodeInfo.java b/src/main/java/weka/finito/structs/NodeInfo.java
index a0891f1..3755912 100644
--- a/src/main/java/weka/finito/structs/NodeInfo.java
+++ b/src/main/java/weka/finito/structs/NodeInfo.java
@@ -20,7 +20,7 @@ public final class NodeInfo implements Serializable, Comparable<NodeInfo> {
 	public final boolean is_leaf;
 	public final String variable_name;
 	public final int comparisonType;
-    public final float threshold;
+    public float threshold;
 
 	private BigInteger paillier;
 	private BigInteger dgk;
@@ -44,6 +44,7 @@ public void encrypt(float threshold, int precision,
 		if (dgk_public_key != null) {
 			this.setDGK(DGKOperations.encrypt(temp_thresh, dgk_public_key));
 		}
+		this.threshold = threshold;
 	}
 
 	public void setDGK(BigInteger dgk){