diff --git a/src/Generator/Java.php b/src/Generator/Java.php
index 9c1a29a..3ab4261 100644
--- a/src/Generator/Java.php
+++ b/src/Generator/Java.php
@@ -57,7 +57,7 @@ protected function writeStruct(Code\Name $name, array $properties, ?string $exte
 
         $discriminator = $origin->getDiscriminator();
         if ($discriminator !== null) {
-            $code.= '@JsonTypeInfo(use = JsonTypeInfo.Id.NAME, include = JsonTypeInfo.As.PROPERTY, property = "' . $discriminator . '")' . "\n";
+            $code.= '@JsonTypeInfo(use = JsonTypeInfo.Id.NAME, property = "' . $discriminator . '")' . "\n";
         }
 
         $mapping = $origin->getMapping();
@@ -153,8 +153,7 @@ protected function writeHeader(DefinitionTypeAbstract $origin, Code\Name $classN
     private function getImports(DefinitionTypeAbstract $origin): array
     {
         $imports = [];
-        $imports[] = 'import com.fasterxml.jackson.annotation.JsonGetter;';
-        $imports[] = 'import com.fasterxml.jackson.annotation.JsonSetter;';
+        $imports[] = 'import com.fasterxml.jackson.annotation.*;';
 
         return $imports;
     }