diff --git a/src/FileUploadExtension.php b/src/FileUploadExtension.php
index 683da11..337814d 100644
--- a/src/FileUploadExtension.php
+++ b/src/FileUploadExtension.php
@@ -87,7 +87,7 @@ public function getConfigSchema(): Schema
 	 */
 	public function afterCompile(\Nette\PhpGenerator\ClassType $class): void
 	{
-		$init = $class->methods['initialize'];
+		$init = $class->getMethods()['initialize'];
 
 		$init->addBody('\Zet\FileUpload\FileUploadControl::register($this->getService(?), ?);', [
 //		$this->getContainerBuilder()->getByType('\Nette\DI\Container'), $this->configuration,