mm: pte: Fix bugs in simple context switch

Signed-off-by: Bart Kuivenhoven <bemkuivenhoven@gmail.com>