diff --git a/research/capsules/__init__.py b/research/capsules/__init__.py
new file mode 100644
index 00000000000..e69de29bb2d
diff --git a/research/capsules/models/layers/layers_test.py b/research/capsules/models/layers/layers_test.py
index fef0a05ed3e..1676cb91b08 100644
--- a/research/capsules/models/layers/layers_test.py
+++ b/research/capsules/models/layers/layers_test.py
@@ -19,6 +19,8 @@
 from __future__ import division
 from __future__ import print_function
 
+import sys
+sys.path.append('../../')
 
 import numpy as np
 import tensorflow as tf