1 files changed, 2 insertions, 2 deletions

diff --git a/drivers/vhost/vhost.c b/drivers/vhost/vhost.c
index c14c42b95ab8..bdb2d6436b2b 100644
--- a/drivers/vhost/vhost.c
+++ b/drivers/vhost/vhost.c
@@ -937,9 +937,9 @@ static int set_bit_to_user(int nr, void __user *addr)
         if (r < 0)
                 return r;
         BUG_ON(r != 1);
-        base = kmap_atomic(page, KM_USER0);
+        base = kmap_atomic(page);
         set_bit(bit, base);
-        kunmap_atomic(base, KM_USER0);
+        kunmap_atomic(base);
         set_page_dirty_lock(page);
         put_page(page);
         return 0;