Fix balance issue