diff --git a/composer.json b/composer.json
index 47c1f27..14cb03f 100644
--- a/composer.json
+++ b/composer.json
@@ -22,13 +22,13 @@
   },
   "require": {
     "php": "^8.1",
-    "nette/utils": "^3.1",
+    "nette/utils": "^3.1|^4.0",
     "nette/php-generator": "^4.0",
     "doctrine/inflector": "^2.0"
   },
   "require-dev": {
     "contributte/qa": "^0.3",
-    "phpunit/phpunit": ">=9.0",
+    "phpunit/phpunit": "^10.0",
     "phpstan/phpstan-deprecation-rules": "^1.0",
     "phpstan/phpstan-strict-rules": "^1.0"
   },
diff --git a/tests/TestEntities/PhpDocPropertyEntity.php b/tests/TestEntities/PhpDocPropertyEntity.php
index a3544b6..e412a84 100644
--- a/tests/TestEntities/PhpDocPropertyEntity.php
+++ b/tests/TestEntities/PhpDocPropertyEntity.php
@@ -1,18 +1,15 @@
-<?php declare(strict_types = 1);
+<?php
 
 namespace DodoIt\EntityGenerator\Tests\TestEntities;
 
-use DodoIt\EntityGenerator\Entity\Entity;
-
 /**
  * @property int $id
  * @property string $title
  * @property int $published
  * @property \DateTimeInterface $created_at
  */
-class PhpDocPropertyEntity extends Entity
+class PhpDocPropertyEntity extends \DodoIt\EntityGenerator\Entity\Entity
 {
-
 	public const TABLE_NAME = 'php_doc_properties';
 
 	public function getId(): int
@@ -20,47 +17,49 @@ public function getId(): int
 		return $this->id;
 	}
 
+
 	public function setId(int $value): self
 	{
 		$this['id'] = $value;
-
 		return $this;
 	}
 
+
 	public function getTitle(): ?string
 	{
 		return $this->title;
 	}
 
+
 	public function setTitle(?string $value): self
 	{
 		$this['title'] = $value;
-
 		return $this;
 	}
 
+
 	public function getPublished(): bool
 	{
 		return $this->published;
 	}
 
+
 	public function setPublished(bool $value): self
 	{
 		$this['published'] = $value;
-
 		return $this;
 	}
 
+
 	public function getCreatedAt(): ?\DateTimeInterface
 	{
 		return $this->created_at;
 	}
 
+
 	public function setCreatedAt(?\DateTimeInterface $value): self
 	{
 		$this['created_at'] = $value;
-
 		return $this;
 	}
-
 }