[PATCH v4 ] sched: fix imbalance flag reset