diff --git a/volia/tsne.py b/volia/tsne.py
index f276c22..abbba8d 100644
--- a/volia/tsne.py
+++ b/volia/tsne.py
@@ -8,7 +8,7 @@ import argparse
 import numpy as np
 from sklearn.manifold import TSNE
 
-from volia.data_io import read_features
+from core.data import read_features
 
 if __name__ == "__main__":
     # Defining argparse