fix return value of decorator