sched: Fix the group_imb logic