diff --git a/darkflow/net/vanilla/train.py b/darkflow/net/vanilla/train.py
index 1785d0407..13de7a454 100644
--- a/darkflow/net/vanilla/train.py
+++ b/darkflow/net/vanilla/train.py
@@ -1,3 +1,5 @@
+import tensorflow as tf
+
 _LOSS_TYPE = ['sse','l2', 'smooth',
 			  'sparse', 'l1', 'softmax',
 			  'svm', 'fisher']
@@ -39,4 +41,4 @@ def loss(self, net_out):
 		assert 'train_size' in m, \
 		'Must specify'
 		size = m['train_size']
-		self.nu = tf.Variable(tf.ones([train_size, num_classes]))
\ No newline at end of file
+		self.nu = tf.Variable(tf.ones([train_size, num_classes]))