Fix schedule while atomic

Signed-off-by: Larry Finger <Larry.Finger@lwfinger.net>