diff --git a/arch/s390/mm/pgtable.c b/arch/s390/mm/pgtable.c
index 9c8a6dd8a49c06b1199f2f1b7ce27db68188f766..71c7eff2c89f0a6489baf3edfd28bdc0e1c228ef 100644
--- a/arch/s390/mm/pgtable.c
+++ b/arch/s390/mm/pgtable.c
@@ -762,7 +762,7 @@ int gmap_ipte_notify(struct gmap *gmap, unsigned long gaddr, unsigned long len)
 			gaddr += PAGE_SIZE;
 			len -= PAGE_SIZE;
 		}
-		spin_unlock(ptl);
+		pte_unmap_unlock(ptep, ptl);
 	}
 	up_read(&gmap->mm->mmap_sem);
 	return rc;