fix return in for