diff --git a/src/main.ts b/src/main.ts
index d389e01..81c8d3a 100644
--- a/src/main.ts
+++ b/src/main.ts
@@ -64,7 +64,7 @@ export default class NodeFactor extends Plugin {
 				if (this.sizeCache.get(node.id) != undefined) {
 					weight = this.sizeCache.get(node.id) as number;
 				} else {
-					calculator.calcNodeWeight(node);
+					weight = calculator.calcNodeWeight(node);
 					this.sizeCache.set(node.id, weight);
 				}
 				node.weight = weight;