1 files changed, 8 insertions, 11 deletions

diff --git a/src/main/java/ChannelNode.java b/src/main/java/ChannelNode.java
index 20a7ac0..6c2e9a3 100644
--- a/src/main/java/ChannelNode.java
+++ b/src/main/java/ChannelNode.java
@@ -43,8 +43,8 @@ public class ChannelNode implements IChannelNode {
         if(dir == null) throw new IllegalArgumentException("dir is null");
 
         Direction.overVals(Map.ofEntries(
-            entry(Direction.INCOMING, () -> incoming.put(node, incoming.get(node) + 1)),
-            entry(Direction.OUTGOING, () -> outgoing.put(node, outgoing.get(node) + 1))
+            entry(Direction.INCOMING, () -> incoming.put(node, ((incoming.get(node) != null) ? incoming.get(node) : 0) + 1)),
+            entry(Direction.OUTGOING, () -> outgoing.put(node, ((outgoing.get(node) != null) ? outgoing.get(node) : 0) + 1))
         ), dir);
     }
 
@@ -56,8 +56,8 @@ public class ChannelNode implements IChannelNode {
 
         for(IChannelNode node: nodes) {
             Direction.overVals(Map.ofEntries(
-                entry(Direction.INCOMING, () -> incoming.put(node, incoming.get(node) + 1)),
-                entry(Direction.OUTGOING, () -> outgoing.put(node, outgoing.get(node) + 1))
+                entry(Direction.INCOMING, () -> incoming.put(node,((incoming.get(node) != null) ? incoming.get(node) : 0) + 1)),
+                entry(Direction.OUTGOING, () -> outgoing.put(node,((outgoing.get(node) != null) ? outgoing.get(node) : 0) + 1))
             ), dir);
         }
     }
@@ -89,13 +89,10 @@ public class ChannelNode implements IChannelNode {
     }
 
     @Override
-    public void clearConnections(Direction dir) throws IllegalArgumentException {
-        if(dir == null) throw new IllegalArgumentException("dir is null");
-
-        Direction.overVals(Map.ofEntries(
-            entry(Direction.INCOMING, () -> incoming.clear()),
-            entry(Direction.OUTGOING, () -> outgoing.clear())
-        ), dir);
+    public void clearConnections() throws IllegalArgumentException {
+        // I have no idea why this is throwing an unsupported operation exception
+        incoming.clear();
+        outgoing.clear();
     }
 
     @Override