diff --git a/mm/reclaim_acct.c b/mm/reclaim_acct.c
index e9fffff3a092a46f4ba49449442a413bafb35aea..18ac0e096c4132c96ea5fee63cb74496514eafac 100644
--- a/mm/reclaim_acct.c
+++ b/mm/reclaim_acct.c
@@ -95,7 +95,6 @@ static void reclaimacct_directreclaim_end(struct reclaim_acct *ra)
 
 	reclaimacct_collect_data();
 	reclaimacct_free(ra, ra->reclaim_type);
-	current->reclaim_acct = NULL;
 }
 
 static void reclaimacct_system_reclaim_end(struct reclaim_acct *ra)
@@ -128,6 +127,8 @@ void reclaimacct_end(enum reclaim_type type)
 		reclaimacct_system_reclaim_end(current->reclaim_acct);
 	else
 		reclaimacct_directreclaim_end(current->reclaim_acct);
+
+	current->reclaim_acct = NULL;
 }
 
 /* Reclaim accounting module initialize */