Fixed imabalance bug