Check for preemption only when returning from interrupt ?