x86: mm: pte: Fix some issues with atomics

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