diff --git a/ann_class/sklearn_ann.py b/ann_class/sklearn_ann.py
index 9c3ce2b4..1feff51f 100644
--- a/ann_class/sklearn_ann.py
+++ b/ann_class/sklearn_ann.py
@@ -17,7 +17,7 @@
 from sklearn.utils import shuffle
 
 # get the data
-X, Y = get_data()
+X, Y, dummy1, dummy2 = get_data()
 
 # split into train and test
 X, Y = shuffle(X, Y)