diff --git a/src/Kdyby/Translation/DI/TranslationExtension.php b/src/Kdyby/Translation/DI/TranslationExtension.php
index e4bfac80..c4341b5b 100644
--- a/src/Kdyby/Translation/DI/TranslationExtension.php
+++ b/src/Kdyby/Translation/DI/TranslationExtension.php
@@ -412,7 +412,7 @@ protected function validateResource($format, $file, $locale, $domain)
 
 		try {
 			$def = $builder->getDefinition($this->loaders[$format]);
-			$refl = Reflection\ClassType::from($def->factory ? $def->factory->entity : $def->class);
+			$refl = Reflection\ClassType::from($def->getEntity() ?: $def->getClass());
 			if (($method = $refl->getConstructor()) && $method->getNumberOfRequiredParameters() > 1) {
 				return;
 			}
@@ -438,7 +438,7 @@ protected function validateResource($format, $file, $locale, $domain)
 
 	public function afterCompile(Code\ClassType $class)
 	{
-		$initialize = $class->methods['initialize'];
+		$initialize = $class->getMethod('initialize');
 		if (class_exists('Tracy\Debugger')) {
 			$initialize->addBody('Kdyby\Translation\Diagnostics\Panel::registerBluescreen();');
 		}