Use control instead of liftBaseWith and restoreM

Just a cosmetic impromevement, since control is the same as liftBaseWith f >>= restoreM.