fix return type to match call