diff --git a/python/draw_net.py b/python/draw_net.py
index cbea5d9ff4e..ba488294275 100755
--- a/python/draw_net.py
+++ b/python/draw_net.py
@@ -5,7 +5,7 @@
 import os
 from google.protobuf import text_format
 
-import caffe
+import caffe, caffe.draw
 from caffe.proto import caffe_pb2
 
 
@@ -17,7 +17,7 @@ def main(argv):
         net = caffe_pb2.NetParameter()
         text_format.Merge(open(sys.argv[1]).read(), net)
         print 'Drawing net to %s' % sys.argv[2]
-        draw_net_to_file(net, sys.argv[2])
+        caffe.draw.draw_net_to_file(net, sys.argv[2])
 
 
 if __name__ == '__main__':