Fix yield() so it works from boot code.