diff --git a/charmcraft/templates/init-django-framework/src/charm.py.j2 b/charmcraft/templates/init-django-framework/src/charm.py.j2
index 96582e0f2..d31095a16 100755
--- a/charmcraft/templates/init-django-framework/src/charm.py.j2
+++ b/charmcraft/templates/init-django-framework/src/charm.py.j2
@@ -14,7 +14,7 @@ import paas_app_charmer.django
 logger = logging.getLogger(__name__)
 
 
-class DjangoCharm(paas_app_charmer.django.Charm):
+class {{ class_name }}(paas_app_charmer.django.Charm):
     """Django Charm service."""
 
     def __init__(self, *args: typing.Any) -> None:
@@ -27,4 +27,4 @@ class DjangoCharm(paas_app_charmer.django.Charm):
 
 
 if __name__ == "__main__":
-    ops.main.main(DjangoCharm)
+    ops.main.main({{ class_name }})
diff --git a/charmcraft/templates/init-flask-framework/src/charm.py.j2 b/charmcraft/templates/init-flask-framework/src/charm.py.j2
index 81fedfff2..b75c440a3 100755
--- a/charmcraft/templates/init-flask-framework/src/charm.py.j2
+++ b/charmcraft/templates/init-flask-framework/src/charm.py.j2
@@ -14,7 +14,7 @@ import paas_app_charmer.flask
 logger = logging.getLogger(__name__)
 
 
-class FlaskCharm(paas_app_charmer.flask.Charm):
+class {{ class_name }}(paas_app_charmer.flask.Charm):
     """Flask Charm service."""
 
     def __init__(self, *args: typing.Any) -> None:
@@ -27,4 +27,4 @@ class FlaskCharm(paas_app_charmer.flask.Charm):
 
 
 if __name__ == "__main__":
-    ops.main.main(FlaskCharm)
+    ops.main.main({{ class_name }})
diff --git a/charmcraft/templates/init-go-framework/src/charm.py.j2 b/charmcraft/templates/init-go-framework/src/charm.py.j2
index 825c204c7..de6162513 100755
--- a/charmcraft/templates/init-go-framework/src/charm.py.j2
+++ b/charmcraft/templates/init-go-framework/src/charm.py.j2
@@ -14,7 +14,7 @@ import paas_app_charmer.go
 logger = logging.getLogger(__name__)
 
 
-class GoCharm(paas_app_charmer.go.Charm):
+class {{ class_name }}(paas_app_charmer.go.Charm):
     """Go Charm service."""
 
     def __init__(self, *args: typing.Any) -> None:
@@ -27,4 +27,4 @@ class GoCharm(paas_app_charmer.go.Charm):
 
 
 if __name__ == "__main__":
-    ops.main.main(GoCharm)
+    ops.main.main({{ class_name }})