diff --git a/src/org/nwapw/abacus/tree/OperatorAssociativity.java b/src/org/nwapw/abacus/function/OperatorAssociativity.java
similarity index 77%
rename from src/org/nwapw/abacus/tree/OperatorAssociativity.java
rename to src/org/nwapw/abacus/function/OperatorAssociativity.java
index 1d7f6df..237b918 100644
--- a/src/org/nwapw/abacus/tree/OperatorAssociativity.java
+++ b/src/org/nwapw/abacus/function/OperatorAssociativity.java
@@ -1,4 +1,4 @@
-package org.nwapw.abacus.tree;
+package org.nwapw.abacus.function;
 
 /**
  * Enum to represent the associativity of an operator.
diff --git a/src/org/nwapw/abacus/tree/TreeNode.java b/src/org/nwapw/abacus/tree/TreeNode.java
index d2d560b..93efc02 100644
--- a/src/org/nwapw/abacus/tree/TreeNode.java
+++ b/src/org/nwapw/abacus/tree/TreeNode.java
@@ -1,5 +1,6 @@
 package org.nwapw.abacus.tree;
 
+import org.nwapw.abacus.function.OperatorAssociativity;
 import org.nwapw.abacus.lexing.Lexer;
 import org.nwapw.abacus.lexing.pattern.Match;