fix balance_test ret struct