[fix] Fixed call of balance function