diff --git a/attributes.npy b/attributes.npy
deleted file mode 100644
index abf10f77bcf949806c0b6bcb54a49bd4eee2b09b..0000000000000000000000000000000000000000
Binary files a/attributes.npy and /dev/null differ
diff --git a/explicability/shap_vals.py b/explicability/shap_vals.py
index d498cdeceb0103c3001447ccec0dc624086e8ce8..f390653609ae1492313a4c443f1db6b00f4211ec 100644
--- a/explicability/shap_vals.py
+++ b/explicability/shap_vals.py
@@ -136,8 +136,7 @@ if __name__ == "__main__":
         "UNDER": "XGB"
     }
     # Retrieve attribute names in order
-    df = pd.read_csv("../gen_train_data/data/input/pre_dataset.csv")
-    attribute_names = list(df.columns.values)
+    attribute_names = list(np.load('../gen_train_data/data/output/attributes.npy', allow_pickle=True))
     # --------------------------------------------------------------------------------------------------------
 
     # Shap value generation