diff --git a/chainer-gogh-multi.py b/chainer-gogh-multi.py
index 32faca0..d92a973 100644
--- a/chainer-gogh-multi.py
+++ b/chainer-gogh-multi.py
@@ -9,7 +9,7 @@
 import chainer
 from chainer import cuda
 import chainer.functions as F
-from chainer.functions import caffe
+from chainer.links import caffe
 from chainer import Variable, optimizers
 
 from models import *
diff --git a/chainer-gogh.py b/chainer-gogh.py
index 2361062..e9da446 100644
--- a/chainer-gogh.py
+++ b/chainer-gogh.py
@@ -10,7 +10,7 @@
 from chainer import cuda
 import chainer.functions as F
 import chainer.links
-from chainer.functions import caffe
+from chainer.links import caffe
 from chainer import Variable, optimizers
 
 from models import *
diff --git a/models.py b/models.py
index c3b6b83..ea54838 100644
--- a/models.py
+++ b/models.py
@@ -2,7 +2,7 @@
 import chainer
 from chainer import cuda
 import chainer.functions as F
-from chainer.functions import caffe
+from chainer.links import caffe
 from chainer import Variable, optimizers