diff --git a/src/pages/3-prediction.py b/src/pages/3-prediction.py
index 7c5e4cfb35edf930c6b7a87b0509ed988ae9248e..6dc97746242bcf8757871935bcfb156954045358 100644
--- a/src/pages/3-prediction.py
+++ b/src/pages/3-prediction.py
@@ -142,7 +142,9 @@ with c2:
                     chem_data, spectra, meta_data, _ = dx_loader(change = hash_)
                     st.success("The data have been loaded successfully", icon="✅")
                     if chem_data.to_numpy().shape[1]>0:
-                        yname = st.selectbox('Select target', options=chem_data.columns, index=chem_data.columns.to_list().index(system_data['data']['target'].name))
+                        # yname = st.selectbox('Select target', options=chem_data.columns, index=chem_data.columns.to_list().index(system_data['data']['target'].name))
+                        yname = system_data['data']['target'].name
+                        st.info("Loaded model to predict " + yname)
                         measured = chem_data.loc[:,yname] == 0
                         y = chem_data.loc[:,yname].loc[measured]
                         pred_data = spectra.loc[measured]