bug fix for gaheft_for_pso