fix return comment