fix return value in test case