Fix balance test