diff --git a/src/MediaManager.html b/src/MediaManager.html
index 96d7905..6784465 100644
--- a/src/MediaManager.html
+++ b/src/MediaManager.html
@@ -1506,7 +1506,7 @@
       builder.addView(view)
         .setOAuthToken(config.token)
         .setDeveloperKey(config.apiKey)
-        .setOrigin(window.location.protocol + '//' + window.location.host)
+        .setOrigin(google.script.host.origin || (window.location.protocol + '//' + window.location.host))
         .setCallback(data => {
           if (data.action == google.picker.Action.PICKED) {
             const doc = data.docs[0];