fix return values in test