diff --git a/xen/common/sched/null.c b/xen/common/sched/null.c
index 2091337fcd..c2dfbb81d0 100644
--- a/xen/common/sched/null.c
+++ b/xen/common/sched/null.c
@@ -551,7 +551,7 @@ static void cf_check null_unit_remove(
 
     cpu = sched_unit_master(unit);
     npc = get_sched_res(cpu)->sched_priv;
-    if ( npc->unit )
+    if ( npc->unit == unit )
         unit_deassign(prv, unit);
 
  out: