fix balance problem