Retain balance after window operations