diff --git a/research/pate_2017/input.py b/research/pate_2017/input.py
index 382787e..ac59b9c 100644
--- a/research/pate_2017/input.py
+++ b/research/pate_2017/input.py
@@ -17,7 +17,7 @@ from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
-from six.moves import pickle
+from six.moves import cPickle as pickle
 import gzip
 import math
 import numpy as np