diff --git a/Source/WebSocketRPC.JS/RPCJs.cs b/Source/WebSocketRPC.JS/RPCJs.cs
index d3aaaeb..25819c6 100644
--- a/Source/WebSocketRPC.JS/RPCJs.cs
+++ b/Source/WebSocketRPC.JS/RPCJs.cs
@@ -128,10 +128,7 @@ public static string GenerateCallerWithDoc<T>(string xmlDocPath, RPCJsSettings<T
         /// <returns>Javascript API.</returns>
         public static string GenerateCallerWithDoc<T>(RPCJsSettings<T> settings = null)
         {
-            var assembly = Assembly.GetEntryAssembly();
-            var fInfo = new FileInfo(assembly.Location);
-
-            var xmlDocPath = Path.ChangeExtension(assembly.Location, ".xml");
+            var xmlDocPath = Path.ChangeExtension(typeof(T).Assembly.Location, ".xml");
 
             if (!File.Exists(xmlDocPath))
                 return GenerateCaller(settings);