Fixed some errors on balancing module