Fix balance check