fix return statement