.

wcfs/internal/xbtree.py

@@ -249,10 +249,14 @@ def Restructure(ztree, newStructure):
             assert not hasattr(rn.node, 'Z')
             rn.node.Z = None
 
+        def D(a, b):
+            return abs(a.range.klo - b.range.klo) + \
+                   abs(a.range.khi - b.range.khi)
+
         # prepare cost matrix
         C = np.zeros((len(RNv), len(RZv)))
-        for j in range(RNv):        # "workers"
-            for i in range(RZv):    # "jobs"
+        for j in range(len(RNv)):       # "workers"
+            for i in range(len(RZv)):   # "jobs"
                 C[j,i] = D(RZv[i], RNv[j])
 
         # find the assignments.