diff --git a/python_modules/libraries/dagster-dg/dagster_dg/templates/COMPONENT_TYPE/COMPONENT_TYPE_NAME_PLACEHOLDER.py.jinja b/python_modules/libraries/dagster-dg/dagster_dg/templates/COMPONENT_TYPE/COMPONENT_TYPE_NAME_PLACEHOLDER.py.jinja index 7bb300ced1339..c441a4df36f4c 100644 --- a/python_modules/libraries/dagster-dg/dagster_dg/templates/COMPONENT_TYPE/COMPONENT_TYPE_NAME_PLACEHOLDER.py.jinja +++ b/python_modules/libraries/dagster-dg/dagster_dg/templates/COMPONENT_TYPE/COMPONENT_TYPE_NAME_PLACEHOLDER.py.jinja @@ -1,12 +1,11 @@ from dagster import Definitions from dagster_components import ( Component, - ComponentRegistry, ComponentLoadContext, - component, + component_type, ) -@component(name="{{ name }}") +@component_type(name="{{ name }}") class {{ component_type_class_name }}(Component): @classmethod