Fix balance method