diff --git a/tree.go b/tree.go
index 05d230c..4523469 100644
--- a/tree.go
+++ b/tree.go
@@ -52,7 +52,7 @@ func TreeFn(treeName string) (TreeConstructorFn, error) {
 	return treeFn, nil
 }
 
-// Remove treeConstructorFN by treeName.
+// removeTreeFn removes a treeConstructorFn by treeName.
 // Only use for test cleanup. Proceed with caution.
 func removeTreeFn(treeName string) {
 	treeFns.Delete(treeName)