diff --git a/generator/ctrl/training_utils/make_forest_tf_records.py b/generator/ctrl/training_utils/make_forest_tf_records.py
index 11217b8..7f93ce9 100644
--- a/generator/ctrl/training_utils/make_forest_tf_records.py
+++ b/generator/ctrl/training_utils/make_forest_tf_records.py
@@ -11,6 +11,9 @@ import argparse
 import fastBPE
 import platform
 import json
+import sys
+# insert at 1, 0 is the script path (or '' in REPL)
+sys.path.insert(1, '../../..')
 from story.utils import *
 
 def make_samples_helper(context, story_block, action_results, path, tree_id):