diff --git a/utils/yangutils/plugin/src/main/java/org/onosproject/yangutils/translator/tojava/YangJavaModelUtils.java b/utils/yangutils/plugin/src/main/java/org/onosproject/yangutils/translator/tojava/YangJavaModelUtils.java
index ff28b66..3489eea 100644
--- a/utils/yangutils/plugin/src/main/java/org/onosproject/yangutils/translator/tojava/YangJavaModelUtils.java
+++ b/utils/yangutils/plugin/src/main/java/org/onosproject/yangutils/translator/tojava/YangJavaModelUtils.java
@@ -147,7 +147,8 @@
                 }
             }
 
-        } else if (javaCodeGeneratorInfo instanceof YangLeavesHolder) {
+        }
+        if (javaCodeGeneratorInfo instanceof YangLeavesHolder) {
             /*
              * Container
              * Case
@@ -189,7 +190,7 @@
      * @throws IOException IO operations fails
      */
     private static void generateTempFiles(JavaCodeGeneratorInfo javaCodeGeneratorInfo,
-                                         YangPluginConfig yangPluginConfig)
+                                          YangPluginConfig yangPluginConfig)
             throws IOException {
         if (!(javaCodeGeneratorInfo instanceof YangNode)) {
             throw new TranslatorException("translation is not supported for the node");
@@ -456,7 +457,7 @@
     /**
      * Returns augment class name.
      *
-     * @param augment YANG augment
+     * @param augment          YANG augment
      * @param yangPluginConfig plugin configurations
      * @return augment class name
      */