diff --git a/python/caffe/__init__.py b/python/caffe/__init__.py
index 0e3808f669528bdd8bf66a0eca7d7bf520ca0ed2..435520fabafc01be0d23f4dc8e32bf932921e1b8 100644
--- a/python/caffe/__init__.py
+++ b/python/caffe/__init__.py
@@ -37,7 +37,7 @@
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 #
 
-import custom_layers
+from . import custom_layers
 from .pycaffe import Net, SGDSolver, NesterovSolver, AdaGradSolver, RMSPropSolver, AdaDeltaSolver, AdamSolver, NCCL, Timer
 from ._caffe import init_log, log, set_mode_cpu, set_mode_gpu, set_device, Layer, get_solver, layer_type_list, set_random_seed, solver_count, set_solver_count, solver_rank, set_solver_rank, set_multiprocess, has_nccl
 from ._caffe import __version__