diff --git a/mm/vmscan.c b/mm/vmscan.c
index 9149444..05d41c0 100644
--- a/mm/vmscan.c
+++ b/mm/vmscan.c
@@ -3372,7 +3372,10 @@
 		}
 	}
 
+	tsk->flags &= ~(PF_MEMALLOC | PF_SWAPWRITE | PF_KSWAPD);
 	current->reclaim_state = NULL;
+	lockdep_clear_current_reclaim_state();
+
 	return 0;
 }