1 files changed, 7 insertions, 1 deletions

diff --git a/virt/kvm/kvm_main.c b/virt/kvm/kvm_main.c
index 162817f853ec..759af6596a07 100644
--- a/virt/kvm/kvm_main.c
+++ b/virt/kvm/kvm_main.c
@@ -679,8 +679,14 @@ static void update_memslots(struct kvm_memslots *slots,
         struct kvm_memory_slot *mslots = slots->memslots;
 
         WARN_ON(mslots[i].id != id);
-        if (!new->npages)
+        if (!new->npages) {
                 new->base_gfn = 0;
+                if (mslots[i].npages)
+                        slots->used_slots--;
+        } else {
+                if (!mslots[i].npages)
+                        slots->used_slots++;
+        }
 
         while (i < KVM_MEM_SLOTS_NUM - 1 &&
                new->base_gfn <= mslots[i + 1].base_gfn) {