diff --git a/betterproto/plugin.py b/betterproto/plugin.py
index 7f2a882..8eecef2 100755
--- a/betterproto/plugin.py
+++ b/betterproto/plugin.py
@@ -183,7 +183,7 @@ def generate_code(request, response):
         lstrip_blocks=True,
         loader=jinja2.FileSystemLoader("%s/templates/" % os.path.dirname(__file__)),
     )
-    template = env.get_template("template.py")
+    template = env.get_template("template.py.j2")
 
     output_map = {}
     for proto_file in request.proto_file:
diff --git a/betterproto/templates/template.py b/betterproto/templates/template.py.j2
similarity index 100%
rename from betterproto/templates/template.py
rename to betterproto/templates/template.py.j2
diff --git a/setup.py b/setup.py
index 191a720..ca1d8c8 100644
--- a/setup.py
+++ b/setup.py
@@ -16,7 +16,7 @@ setup(
     packages=find_packages(
         exclude=["tests", "*.tests", "*.tests.*", "output", "output.*"]
     ),
-    package_data={"betterproto": ["py.typed", "templates/template.py"]},
+    package_data={"betterproto": ["py.typed", "templates/template.py.j2"]},
     python_requires=">=3.6",
     install_requires=[
         'dataclasses; python_version<"3.7"',