diff --git a/docs/index.rst b/docs/index.rst
index 1f2a624a0..b9768f621 100644
--- a/docs/index.rst
+++ b/docs/index.rst
@@ -48,7 +48,8 @@ You can render an Airflow Task Group using the ``DbtTaskGroup`` class. Here's an
 
     from airflow import DAG
     from airflow.operators.empty import EmptyOperator
-    from cosmos.task_group import DbtTaskGroup
+    from cosmos import DbtTaskGroup
+
 
     profile_config = ProfileConfig(
         profile_name="default",