diff --git a/hybnet_train_fun.m b/hybnet_train_fun.m
index 61f0546..40a3fa9 100644
--- a/hybnet_train_fun.m
+++ b/hybnet_train_fun.m
@@ -180,7 +180,7 @@
     [~]=hnetsimul(hnet,data,iplot,1);
     
     save([simfile '.mat'], 'hnet', 'data','wfinal','runs','nruns','ind')
-    
+    data=runs.data_iter{1, ind};
     
 else
     load([simfile '.mat']);
@@ -430,4 +430,4 @@
 
 % save data_temp.mat data
 
-end
\ No newline at end of file
+end