diff --git a/compiler/base/parser/src/main/resources/GeneratedYang.g4 b/compiler/base/parser/src/main/resources/GeneratedYang.g4
index af16c05..0646a60 100644
--- a/compiler/base/parser/src/main/resources/GeneratedYang.g4
+++ b/compiler/base/parser/src/main/resources/GeneratedYang.g4
@@ -21,10 +21,6 @@
 grammar GeneratedYang;
 import YangLexer;
 
-@header {
-package org.onosproject.yang.compiler.parser.antlrgencode;
-}
-
     yangfile : moduleStatement EOF
              | subModuleStatement EOF;
 
diff --git a/compiler/base/parser/src/main/resources/YangLexer.g4 b/compiler/base/parser/src/main/resources/YangLexer.g4
index 51a9231..45189f7 100644
--- a/compiler/base/parser/src/main/resources/YangLexer.g4
+++ b/compiler/base/parser/src/main/resources/YangLexer.g4
@@ -20,6 +20,10 @@
 
 lexer grammar YangLexer;
 
+@header {
+package org.onosproject.yang.compiler.parser.antlrgencode;
+}
+
     // Statements keywords
     ANYXML_KEYWORD      : 'anyxml';
     ARGUMENT_KEYWORD    : 'argument';