diff --git a/mnist1d/data.py b/mnist1d/data.py
index 3f5248b..fa1c551 100644
--- a/mnist1d/data.py
+++ b/mnist1d/data.py
@@ -4,6 +4,7 @@
 import numpy as np
 import os
 import requests
+import mnist1d
 from mnist1d.transform import transform
 from mnist1d.utils import from_pickle, to_pickle, ObjectView, set_seed