diff --git a/module/nvpair/nvpair.c b/module/nvpair/nvpair.c
index 5c6898446793..7c8a5d99f6f0 100644
--- a/module/nvpair/nvpair.c
+++ b/module/nvpair/nvpair.c
@@ -271,7 +271,7 @@ nvlist_alloc(nvlist_t **nvlp, uint_t nvflag, int kmflag)
 {
 #if defined(_KERNEL) && !defined(_BOOT)
 	return (nvlist_xalloc(nvlp, nvflag,
-	    (kmflag == KM_SLEEP ? nv_alloc_sleep : nv_alloc_nosleep)));
+	    (kmflag & __GFP_WAIT ? nv_alloc_sleep : nv_alloc_nosleep)));
 #else
 	return (nvlist_xalloc(nvlp, nvflag, nv_alloc_nosleep));
 #endif