Fix balance command