diff --git a/dibi/Nette/DibiNetteExtension.php b/dibi/Nette/DibiNetteExtension.php
index a19f5803..90721065 100644
--- a/dibi/Nette/DibiNetteExtension.php
+++ b/dibi/Nette/DibiNetteExtension.php
@@ -23,7 +23,7 @@ class DibiNetteExtension extends Nette\Config\CompilerExtension
 
 	public function loadConfiguration()
 	{
-		$container = $this->getContainer();
+		$container = $this->getContainerBuilder();
 		$config = $this->getConfig();
 
 		$useProfiler = isset($config['profiler'])