fix "wait" to happen only when affinity is correct (oops)