diff --git a/src/Api/InterfaceControllerProvider.php b/src/Api/InterfaceControllerProvider.php
index 1f91fca..ac0268e 100644
--- a/src/Api/InterfaceControllerProvider.php
+++ b/src/Api/InterfaceControllerProvider.php
@@ -27,7 +27,7 @@ class InterfaceControllerProvider implements ControllerProviderInterface
         $controllers->get('/images', function (Application $app) {
 
             $getEmoBlocks = function($emotion, $feature, $steps = 6) use($em) {
-                $hitRepo = $this->_em->getEm()->getRepository(EmotionHero\Models\Hit::class);
+                $hitRepo = $this->_eh->getEm()->getRepository(EmotionHero\Models\Hit::class);
                 $blocks = [];
                 foreach(range(0, 100, 100/($steps-1)) as $i) {
                     /* @var $hitRepo EmotionHero\Models\HitRepository */
@@ -61,7 +61,7 @@ class InterfaceControllerProvider implements ControllerProviderInterface
             return $app['serializer']->serialize($output, 'json');
         });
 
-        
+
         return $controllers;
     }
 }