1 files changed, 2 insertions, 4 deletions

diff --git a/include/linux/namespace.h b/include/linux/namespace.h
index 3abc8e3b4879..d137009f0b2b 100644
--- a/include/linux/namespace.h
+++ b/include/linux/namespace.h
@@ -4,6 +4,7 @@
 
 #include <linux/mount.h>
 #include <linux/sched.h>
+#include <linux/nsproxy.h>
 
 struct namespace {
         atomic_t                count;
@@ -26,11 +27,8 @@ static inline void put_namespace(struct namespace *namespace)
 
 static inline void exit_namespace(struct task_struct *p)
 {
-        struct namespace *namespace = p->namespace;
+        struct namespace *namespace = p->nsproxy->namespace;
         if (namespace) {
-                task_lock(p);
-                p->namespace = NULL;
-                task_unlock(p);
                 put_namespace(namespace);
         }
 }