diff --git a/setup.py b/setup.py
index <HASH>..<HASH> 100644
--- a/setup.py
+++ b/setup.py
@@ -29,4 +29,7 @@ setup(
     ],
     keywords='dataset data iteration pipeline processing',
     packages=find_packages(exclude=['tests']),
-    install_requires=['six'])
+    install_requires=['six', 'picklable_itertools'],
+    dependency_links=['http://github.com/dwf/picklable_itertools/'
+                      'tarball/master#egg=picklable_itertools'],
+    )
diff --git a/tests/test_mnist.py b/tests/test_mnist.py
index <HASH>..<HASH> 100644
--- a/tests/test_mnist.py
+++ b/tests/test_mnist.py
@@ -1,8 +1,6 @@
 import numpy
 from numpy.testing import assert_raises
 
-import theano
-
 from fuel.mnist import MNIST
 
 
@@ -18,7 +16,7 @@ def test_mnist():
 
     first_feature, first_target = mnist_train.get_data(request=[0])
     assert first_feature.shape == (1, 784)
-    assert first_feature.dtype is numpy.dtype(theano.config.floatX)
+    assert first_feature.dtype.kind == 'f'
     assert first_target.shape == (1, 1)
     assert first_target.dtype is numpy.dtype('uint8')
Add picklable_itertools to dependencies