runtime: add some more preemption checks