diff --git a/docs/how-to-guides/develop-a-tool/create-your-own-custom-strong-type-connection.md b/docs/how-to-guides/develop-a-tool/create-your-own-custom-strong-type-connection.md
index 1dda32d36f3a..f5ffeba4d634 100644
--- a/docs/how-to-guides/develop-a-tool/create-your-own-custom-strong-type-connection.md
+++ b/docs/how-to-guides/develop-a-tool/create-your-own-custom-strong-type-connection.md
@@ -13,7 +13,7 @@ For other connections types, please refer to [Connections](https://microsoft.git
 
   ```bash
   # Eventually only need to pip install promptflow==0.1.0b8
-  pip install "promptflow==0.0.108028960" --extra-index-url https://azuremlsdktestpypi.azureedge.net/promptflow
+  pip install "promptflow==0.0.108377202" --extra-index-url https://azuremlsdktestpypi.azureedge.net/promptflow
   ```
 
 ## Create your own custom strong type connection
@@ -56,6 +56,7 @@ Make sure that you adhere to the following guidelines:
   The purpose of parsing the docstring into a connection template is to enhance the clarity and understanding for users when creating a custom strong type connection. By providing detailed explanations for each key, users can easily grasp the meaning and purpose of each key, allowing them to accurately fill in the corresponding values.
   
   ```yaml
+  $schema: https://azuremlschemas.azureedge.net/promptflow/latest/CustomStrongTypeConnection.schema.json
   name: "to_replace_with_connection_name"
   type: custom
   custom_type: MyCustomConnection
diff --git a/examples/flows/standard/flow-with-package-tool-using-custom-strong-type-connection/my_custom_connection.yml b/examples/flows/standard/flow-with-package-tool-using-custom-strong-type-connection/my_custom_connection.yml
index 59621280525c..2d7d2194e032 100644
--- a/examples/flows/standard/flow-with-package-tool-using-custom-strong-type-connection/my_custom_connection.yml
+++ b/examples/flows/standard/flow-with-package-tool-using-custom-strong-type-connection/my_custom_connection.yml
@@ -1,3 +1,4 @@
+$schema: https://azuremlschemas.azureedge.net/promptflow/latest/CustomStrongTypeConnection.schema.json
 name: "my_custom_connection"
 type: custom
 custom_type: MyCustomConnection