Fix some error, context_switch still doesn't work.