diff --git a/src/Microsoft.AspNet.Diagnostics/RuntimeInfoMiddleware.cs b/src/Microsoft.AspNet.Diagnostics/RuntimeInfoMiddleware.cs
index de3ee8d286..35d0fb5445 100644
--- a/src/Microsoft.AspNet.Diagnostics/RuntimeInfoMiddleware.cs
+++ b/src/Microsoft.AspNet.Diagnostics/RuntimeInfoMiddleware.cs
@@ -1,15 +1,13 @@
 // Copyright (c) Microsoft Open Technologies, Inc. All rights reserved.
 // Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
 
-using System;
+using System.Reflection;
 using System.Threading.Tasks;
 using Microsoft.AspNet.Builder;
 using Microsoft.AspNet.Diagnostics.Views;
 using Microsoft.AspNet.Http;
-using Microsoft.Framework.Runtime;
-using System.Linq;
-using System.Reflection;
 using Microsoft.Framework.Internal;
+using Microsoft.Framework.Runtime;
 
 namespace Microsoft.AspNet.Diagnostics
 {
@@ -64,11 +62,12 @@ namespace Microsoft.AspNet.Diagnostics
             return model;
         }
 
-        private static string GetRuntimeVersion()
+        private string GetRuntimeVersion()
         {
-            var klr = Assembly.Load(new AssemblyName("dnx.host"));
-            var version = klr.GetCustomAttribute<AssemblyInformationalVersionAttribute>();
+            var version = _libraryManager
+                .GetType().GetTypeInfo().Assembly
+                .GetCustomAttribute<AssemblyInformationalVersionAttribute>();
             return version?.InformationalVersion;
         }
     }
-}
+}
\ No newline at end of file