diff --git a/raiss/windows.py b/raiss/windows.py
index 120981fbd231f8e2acce09c5387e2e9d29c66914..707bd0dd4478c616cd562f6edb66479102bf85f6 100644
--- a/raiss/windows.py
+++ b/raiss/windows.py
@@ -188,7 +188,8 @@ def impg_like_imputation(ld_file, ref_panel, zscore, window_size, buffer, lamb,
 
                 # keep only SNP with non negligible explained variance
                 snp_well_predicted = (batch_df.Var < 0.9)
-                batch_df_filt = batch_df.reindex[in_core_window & snp_well_predicted, zscore_results.columns]
+                print(batch_df)
+                batch_df_filt = batch_df.loc[(in_core_window & snp_well_predicted), zscore_results.columns]
                 zscore_results = pd.concat([zscore_results, batch_df_filt])
             else:
                 zscore_results = pd.concat([zscore_results, zt])