diff --git a/data-upgrade-pages/src/test/java/org/exoplatform/migration/SpaceApplicationMigrationTest.java b/data-upgrade-pages/src/test/java/org/exoplatform/migration/SpaceApplicationMigrationTest.java
index 270fd0a56..c8813b903 100644
--- a/data-upgrade-pages/src/test/java/org/exoplatform/migration/SpaceApplicationMigrationTest.java
+++ b/data-upgrade-pages/src/test/java/org/exoplatform/migration/SpaceApplicationMigrationTest.java
@@ -22,6 +22,11 @@
 import org.exoplatform.social.core.space.*;
 import org.exoplatform.social.core.space.model.Space;
 import org.exoplatform.social.core.space.spi.SpaceService;
+import org.gatein.pc.api.Portlet;
+import org.gatein.pc.api.PortletContext;
+import org.gatein.pc.api.PortletInvoker;
+import org.gatein.pc.api.PortletInvokerException;
+import org.gatein.pc.api.info.PortletInfo;
 import org.junit.After;
 import org.junit.Before;
 import org.junit.Test;
@@ -29,9 +34,12 @@
 import org.mockito.junit.MockitoJUnitRunner;
 
 
+import java.util.Collections;
 import java.util.HashSet;
 
 import static org.junit.Assert.*;
+import static org.mockito.Mockito.mock;
+import static org.mockito.Mockito.when;
 
 @RunWith(MockitoJUnitRunner.class)
 @ConfiguredBy({
@@ -60,13 +68,33 @@ public SpaceApplicationMigrationTest() {
   }
 
   @Before
-  public void setUp() {
+  public void setUp() throws PortletInvokerException {
     container = getContainer();
     this.spaceService = container.getComponentInstanceOfType(SpaceService.class);
     this.navigationService = container.getComponentInstanceOfType(NavigationService.class);
     this.identityRegistry = container.getComponentInstanceOfType(IdentityRegistry.class);
     this.entityManagerService = container.getComponentInstanceOfType(EntityManagerService.class);
     this.settingService = container.getComponentInstanceOfType(SettingService.class);
+    PortletInvoker portletInvoker = mock(PortletInvoker.class);
+    Portlet portlet = new Portlet() {
+      @Override
+      public PortletContext getContext() {
+        return null;
+      }
+
+      @Override
+      public PortletInfo getInfo() {
+        return null;
+      }
+
+      @Override
+      public boolean isRemote() {
+        return false;
+      }
+    };
+    when(portletInvoker.getPortlets()).thenReturn(Collections.singleton(portlet));
+    container.unregisterComponent(PortletInvoker.class);
+    container.registerComponentInstance(PortletInvoker.class, portletInvoker);
     HashSet<MembershipEntry> memberships = new HashSet<MembershipEntry>();
     memberships.add(new MembershipEntry("/platform/users", "*"));
     memberships.add(new MembershipEntry("/platform/administrators", "*"));