diff --git a/javascript/src/viamapi-iframe.js b/javascript/src/viamapi-iframe.js
index b4060b63e40c227c32373dd079cda2a5bdcb8dd9..073f9850c531a98b24f1e1ec0c155735583c65b7 100644
--- a/javascript/src/viamapi-iframe.js
+++ b/javascript/src/viamapi-iframe.js
@@ -2266,6 +2266,8 @@ const connection = Penpal.connectToParent({
 connection.promise.then(parent => {
   var identities = localStorage.getItem("identities")
 
+  window.API_HOST = `${window.location.protocol}//${window.location.host}/`;
+
   console.log("Library loaded at: " + new Date().toISOString())
 
   if (identities === "" || identities === null) {
diff --git a/main.go b/main.go
index 70dd5a6bda6a77c14fbff00ae091d759cb13799e..cb7a9ac09702f42cc24b1d8676e2687aa6cfe28e 100644
--- a/main.go
+++ b/main.go
@@ -29,10 +29,8 @@ func GetClientJsLibrary(iframeUrl string) string {
 	return strings.Replace(string(dat), "{{urlArg}}", iframeUrl, 1) + "\n"
 }
 
-func GetIframeJsLibrary(host string,
-	endPoints map[string]*types.EndPoint) string {
+func GetIframeJsLibrary(endPoints map[string]*types.EndPoint) string {
 	result := "<script>\n\n"
-	result += "window.API_HOST = '" + host + "';\n\n"
 
 	var keys []string
 	for k := range endPoints {