diff --git a/api/src/main/java/ai/djl/modality/cv/translator/ImageClassificationTranslatorFactory.java b/api/src/main/java/ai/djl/modality/cv/translator/ImageClassificationTranslatorFactory.java
index 6f9a6b2e709..9ee08ceb501 100644
--- a/api/src/main/java/ai/djl/modality/cv/translator/ImageClassificationTranslatorFactory.java
+++ b/api/src/main/java/ai/djl/modality/cv/translator/ImageClassificationTranslatorFactory.java
@@ -51,12 +51,14 @@ public class ImageClassificationTranslatorFactory
         EXPANSIONS.put(new Pair<>(Input.class, Output.class), ImageServingTranslator::new);
     }
 
+    /** {@inheritDoc} */
     @Override
     protected Translator<Image, Classifications> buildBaseTranslator(
             Model model, Map<String, ?> arguments) {
         return ImageClassificationTranslator.builder(arguments).build();
     }
 
+    /** {@inheritDoc} */
     @Override
     protected Map<Pair<Type, Type>, Function<Translator<Image, Classifications>, Translator<?, ?>>>
             getExpansions() {
diff --git a/api/src/main/java/ai/djl/translate/ExpansionTranslatorFactory.java b/api/src/main/java/ai/djl/translate/ExpansionTranslatorFactory.java
index 4631adf982d..952f2901aec 100644
--- a/api/src/main/java/ai/djl/translate/ExpansionTranslatorFactory.java
+++ b/api/src/main/java/ai/djl/translate/ExpansionTranslatorFactory.java
@@ -43,9 +43,8 @@ public <I, O> Translator<I, O> newInstance(
         return newInstance(input, output, baseTranslator);
     }
 
-    /** {@inheritDoc} */
     @SuppressWarnings("unchecked")
-    public <I, O> Translator<I, O> newInstance(
+    <I, O> Translator<I, O> newInstance(
             Class<I> input, Class<O> output, Translator<IbaseT, ObaseT> translator) {
         Function<Translator<IbaseT, ObaseT>, Translator<?, ?>> expansion =
                 getExpansions().get(new Pair<>(input, output));
@@ -84,17 +83,20 @@ protected abstract Translator<IbaseT, ObaseT> buildBaseTranslator(
             getExpansions();
 
     final class ExpandedTranslatorOptions implements TranslatorOptions {
+
         private Translator<IbaseT, ObaseT> translator;
 
         private ExpandedTranslatorOptions(Translator<IbaseT, ObaseT> translator) {
             this.translator = translator;
         }
 
+        /** {@inheritDoc} */
         @Override
         public Set<Pair<Type, Type>> getOptions() {
             return getSupportedTypes();
         }
 
+        /** {@inheritDoc} */
         @Override
         public <I, O> Translator<I, O> option(Class<I> input, Class<O> output) {
             return newInstance(input, output, translator);