1 files changed, 6 insertions, 1 deletions
diff --git a/lib/cli.py b/lib/cli.py
index b6ccbc2..4c56e73 100644
--- a/lib/cli.py
+++ b/lib/cli.py
@@ -1,6 +1,11 @@
 #!/usr/bin/env python3
 
-from dfatool.functions import SplitFunction, AnalyticFunction, StaticFunction
+from dfatool.functions import (
+    SplitFunction,
+    AnalyticFunction,
+    StaticFunction,
+    FOLFunction,
+)
 
 
 def print_static(model, static_model, name, attribute):