diff --git a/java_gen/codegen.py b/java_gen/codegen.py
index 258b81c..07b910a 100644
--- a/java_gen/codegen.py
+++ b/java_gen/codegen.py
@@ -74,6 +74,7 @@
 
         context['class_name'] = clazz.name
         context['package'] = clazz.package
+        context['template_dir'] = self.templates_dir
 
         filename = os.path.join(self.basedir, src_dir, "%s/%s.java" % (clazz.package.replace(".", "/"), clazz.name))
         dirname = os.path.dirname(filename)