Fixes running in the iframe