fix ps_correl for new API