ContainerBuilder: used real container's typehint in generated factories

nette · Mar 10, 2015 · c0cf062 · c0cf062
1 parent 3f9490a
commit c0cf062
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/DI/ContainerBuilder.php b/src/DI/ContainerBuilder.php
@@ -519,7 +519,7 @@ private function generateService($name)
 		$factoryClass->addMethod('__construct')
 			->addBody('$this->container = $container;')
 			->addParameter('container')
-				->setTypeHint('Nette\DI\Container');
+				->setTypeHint($this->generatedClasses[0]->name);
 
 		$factoryClass->addMethod($def->implementType)
 			->setParameters($this->convertParameters($def->parameters))