diff --git a/src/API/TfRnnCell.jl b/src/API/TfRnnCell.jl
index 283d53d..bffe518 100644
--- a/src/API/TfRnnCell.jl
+++ b/src/API/TfRnnCell.jl
@@ -3,7 +3,7 @@
 module TfRnnCell
 using PyCall
 @pyimport tensorflow as tf
-@pyimport tensorflow.models.rnn.rnn_cell as tf_rnn_cell
+@pyimport tensorflow.python.ops.rnn_cell as tf_rnn_cell
 import TensorFlow.CoreTypes: *
 using TensorFlow.CoreTypes
 
diff --git a/src/Models.jl b/src/Models.jl
index 40f3465..6f5a4dd 100644
--- a/src/Models.jl
+++ b/src/Models.jl
@@ -4,7 +4,7 @@ module RNN
 using PyCall
 export LSTMCell
 
-@pyimport tensorflow.models.rnn.rnn_cell as rnn_cell
+@pyimport tensorflow.python.ops.rnn_cell as rnn_cell
 
 immutable LSTMCell
   x::PyObject