1 files changed, 2 insertions, 1 deletions

diff --git a/virt/kvm/eventfd.c b/virt/kvm/eventfd.c
index 62e4cd947a90..a9d3fc6c681c 100644
--- a/virt/kvm/eventfd.c
+++ b/virt/kvm/eventfd.c
@@ -72,12 +72,13 @@ static void
 irqfd_shutdown(struct work_struct *work)
 {
         struct _irqfd *irqfd = container_of(work, struct _irqfd, shutdown);
+        u64 cnt;
 
         /*
          * Synchronize with the wait-queue and unhook ourselves to prevent
          * further events.
          */
-        remove_wait_queue(irqfd->wqh, &irqfd->wait);
+        eventfd_ctx_remove_wait_queue(irqfd->eventfd, &irqfd->wait, &cnt);
 
         /*
          * We know no new events will be scheduled at this point, so block