--- x/mm/mmap_lock.c
+++ y/mm/mmap_lock.c
@@ -240,6 +240,7 @@ retry:
 		/* Check if the VMA got isolated after we found it */
 		if (PTR_ERR(vma) == -EAGAIN) {
 			count_vm_vma_lock_event(VMA_LOCK_MISS);
+			mas_set(&mas, address);
 			/* The area was replaced with another one */
 			goto retry;
 		}