diff --git a/theano_rhn.py b/theano_rhn.py
index fcdb29e..b23a3ee 100644
--- a/theano_rhn.py
+++ b/theano_rhn.py
@@ -1,7 +1,11 @@
 from __future__ import absolute_import, division, print_function
 
 import numbers
-import cPickle
+
+try:
+  import cPickle
+except ImportError:
+  import _pickle
 
 import numpy as np