diff --git a/run_sage.sh b/run_sage.sh
new file mode 100755
index 0000000..9eeadc7
--- /dev/null
+++ b/run_sage.sh
@@ -0,0 +1,2 @@
+#!/bin/bash
+PYTHONPATH=src sage src/tea3/tea3model.py
diff --git a/src/tea3/tea3model.py b/src/tea3/tea3model.py
index 95d86f8..b472cf0 100644
--- a/src/tea3/tea3model.py
+++ b/src/tea3/tea3model.py
@@ -1,5 +1,8 @@
 from sage.all import *
 
+from tea3.constants import TEA3_SBOX, T_F1, T_F2
+
+
 F = GF(2)
 
 names = (