diff --git a/optim_nn.py b/optim_nn.py
index 609b654..66fd8fa 100755
--- a/optim_nn.py
+++ b/optim_nn.py
@@ -7,7 +7,7 @@
 #           this is gonna take some work.
 
 from optim_nn_core import *
-from optim_nn_core import _check, _f
+from optim_nn_core import _check, _f, _0, _1
 
 import sys