diff --git a/jest.config.itl.js b/jest.config.itl.js
index e74e9c58..0c03898a 100644
--- a/jest.config.itl.js
+++ b/jest.config.itl.js
@@ -16,4 +16,5 @@ module.exports = {
   moduleNameMapper: {
     'app/main': FALLBACK_APP_MAIN_PATH,
   },
+  setupFiles: ['<rootDir>/src/main.ts'], // To ensure that pluginLoader is called
 };
diff --git a/packages/plugin-atoms/src/components/__tests__/IframeSpec.jsx b/packages/plugin-atoms/src/components/__tests__/IframeSpec.jsx
index 05877b44..b020e25e 100644
--- a/packages/plugin-atoms/src/components/__tests__/IframeSpec.jsx
+++ b/packages/plugin-atoms/src/components/__tests__/IframeSpec.jsx
@@ -1,6 +1,7 @@
 import { renderWithContext, waitFor } from '@ima/testing-library';
 
-import { Iframe, Visibility } from '../../main';
+import Visibility from '../../Visibility';
+import { Iframe } from '../Iframe';
 
 describe('Iframe', () => {
   it('should render with noscript tag by default', async () => {
diff --git a/packages/plugin-atoms/src/components/__tests__/ImageSpec.jsx b/packages/plugin-atoms/src/components/__tests__/ImageSpec.jsx
index a907eb17..57b9502a 100644
--- a/packages/plugin-atoms/src/components/__tests__/ImageSpec.jsx
+++ b/packages/plugin-atoms/src/components/__tests__/ImageSpec.jsx
@@ -1,5 +1,3 @@
-import '../../main'; // To ensure that pluginLoader is called
-
 import { renderWithContext } from '@ima/testing-library';
 
 import { Image } from '../Image';