fix return keyword bug