Add tests for the iframe fix