diff --git a/src/Microsoft.AspNet.WebSockets.Protocol/project.json b/src/Microsoft.AspNet.WebSockets.Protocol/project.json
index a0fce33e1d..ae5cd3be66 100644
--- a/src/Microsoft.AspNet.WebSockets.Protocol/project.json
+++ b/src/Microsoft.AspNet.WebSockets.Protocol/project.json
@@ -2,7 +2,7 @@
     "version": "1.0.0-*",
     "description": "Managed web socket protocol parser.",
     "dependencies": {
-        "Microsoft.Framework.NotNullAttribute.Internal": { "type": "build", "version": "1.0.0-*" }
+        "Microsoft.Framework.NotNullAttribute.Sources": { "type": "build", "version": "1.0.0-*" }
     },
     "frameworks": {
         "net45": { },
diff --git a/test/Microsoft.AspNet.WebSockets.Protocol.Test/project.json b/test/Microsoft.AspNet.WebSockets.Protocol.Test/project.json
index 86111c50e7..7db25a7abe 100644
--- a/test/Microsoft.AspNet.WebSockets.Protocol.Test/project.json
+++ b/test/Microsoft.AspNet.WebSockets.Protocol.Test/project.json
@@ -1,7 +1,7 @@
 {
     "dependencies": {
         "Microsoft.AspNet.WebSockets.Protocol": "1.0.0-*",
-        "Microsoft.Framework.NotNullAttribute.Internal": { "type": "build", "version": "1.0.0-*" },
+        "Microsoft.Framework.NotNullAttribute.Sources": { "type": "build", "version": "1.0.0-*" },
         "xunit.runner.aspnet": "2.0.0-aspnet-*"
     },
     "frameworks": {