diff --git a/website/www/site/content/en/documentation/transforms/python/aggregation/cogroupbykey.md b/website/www/site/content/en/documentation/transforms/python/aggregation/cogroupbykey.md
index d5dac4ed9d73..b8594823d7ae 100644
--- a/website/www/site/content/en/documentation/transforms/python/aggregation/cogroupbykey.md
+++ b/website/www/site/content/en/documentation/transforms/python/aggregation/cogroupbykey.md
@@ -38,17 +38,9 @@ Then, we apply `CoGroupByKey` to join both `PCollection`s using their keys.
 `CoGroupByKey` expects a dictionary of named keyed `PCollection`s, and produces elements joined by their keys.
 The values of each output element are dictionaries where the names correspond to the input dictionary, with lists of all the values found for that key.
 
-{{< highlight language="py" py="sdks/python/apache_beam/examples/snippets/transforms/aggregation/cogroupbykey.py" >}}
-{{< code_sample "sdks/python/apache_beam/examples/snippets/transforms/aggregation/cogroupbykey.py" cogroupbykey >}}
-{{< /highlight >}}
-
-{{< paragraph class="notebook-skip" >}}
-Output:
-{{< /paragraph >}}
-
-{{< highlight class="notebook-skip" >}}
-{{< code_sample "sdks/python/apache_beam/examples/snippets/transforms/aggregation/cogroupbykey_test.py" plants >}}
-{{< /highlight >}}
+{{< playground height="800px" >}}
+{{< playground_snippet language="py" path="SDK_PYTHON_CoGroupByKeyMerge" show="cogroupbykey" >}}
+{{< /playground >}}
 
 ## Related transforms