fix return signature