--- x/mm/filemap.c
+++ y/mm/filemap.c
@@ -3439,7 +3439,6 @@ retry_find:
 		folio_put(folio);
 		goto retry_find;
 	}
-	VM_BUG_ON_FOLIO(!folio_contains(folio, index), folio);
 
 	/*
 	 * We have a locked folio in the page cache, now we need to check
@@ -3490,6 +3489,7 @@ retry_find:
 		return VM_FAULT_SIGBUS;
 	}
 
+	VM_BUG_ON_FOLIO(!folio_contains(folio, index), folio);
 	vmf->page = folio_file_page(folio, index);
 	return ret | VM_FAULT_LOCKED;