Revert "mm: simplify code of swap.c"