diff --git a/src/DoctrineDataFixtureModule/Module.php b/src/DoctrineDataFixtureModule/Module.php
index a229110..311dc3c 100644
--- a/src/DoctrineDataFixtureModule/Module.php
+++ b/src/DoctrineDataFixtureModule/Module.php
@@ -68,7 +68,7 @@ public function init(ModuleManager $e)
 
             /* @var $sm ServiceLocatorInterface */
             $sm = $e->getParam('ServiceManager');
-            $em = $sm->get('doctrine.entitymanager.orm_default');
+            $em = $cli->getHelperSet()->get('em')->getEntityManager();
             $paths = $sm->get('doctrine.configuration.fixtures');
 
             $importCommand = new ImportCommand($sm);