diff --git a/aptos-move/framework/aptos-framework/doc/dkg.md b/aptos-move/framework/aptos-framework/doc/dkg.md
index 9db68a09dd5d3..09b667fea28ef 100644
--- a/aptos-move/framework/aptos-framework/doc/dkg.md
+++ b/aptos-move/framework/aptos-framework/doc/dkg.md
@@ -7,10 +7,12 @@ DKG on-chain states and helper functions.
 
 
 -  [Struct `DKGSessionMetadata`](#0x1_dkg_DKGSessionMetadata)
+-  [Enum `TestEnum`](#0x1_dkg_TestEnum)
 -  [Struct `DKGStartEvent`](#0x1_dkg_DKGStartEvent)
 -  [Struct `DKGSessionState`](#0x1_dkg_DKGSessionState)
 -  [Resource `DKGState`](#0x1_dkg_DKGState)
 -  [Constants](#@Constants_0)
+-  [Function `test_a`](#0x1_dkg_test_a)
 -  [Function `initialize`](#0x1_dkg_initialize)
 -  [Function `start`](#0x1_dkg_start)
 -  [Function `finish`](#0x1_dkg_finish)
@@ -80,6 +82,55 @@ This can be considered as the public input of DKG.
 </dl>
 
 
+</details>
+
+<a id="0x1_dkg_TestEnum"></a>
+
+## Enum `TestEnum`
+
+
+
+<pre><code>enum <a href="dkg.md#0x1_dkg_TestEnum">TestEnum</a> <b>has</b> drop
+</code></pre>
+
+
+
+<details>
+<summary>Variants</summary>
+
+
+<details>
+<summary>A</summary>
+
+
+<details>
+<summary>Fields</summary>
+
+
+<dl>
+</dl>
+
+
+</details>
+
+</details>
+
+<details>
+<summary>B</summary>
+
+
+<details>
+<summary>Fields</summary>
+
+
+<dl>
+</dl>
+
+
+</details>
+
+</details>
+
 </details>
 
 <a id="0x1_dkg_DKGStartEvent"></a>
@@ -214,6 +265,30 @@ The completed and in-progress DKG sessions.
 
 
 
+<a id="0x1_dkg_test_a"></a>
+
+## Function `test_a`
+
+
+
+<pre><code><b>fun</b> <a href="dkg.md#0x1_dkg_test_a">test_a</a>(): <a href="dkg.md#0x1_dkg_TestEnum">dkg::TestEnum</a>
+</code></pre>
+
+
+
+<details>
+<summary>Implementation</summary>
+
+
+<pre><code><b>fun</b> <a href="dkg.md#0x1_dkg_test_a">test_a</a>(): <a href="dkg.md#0x1_dkg_TestEnum">TestEnum</a> {
+   TestEnum::A
+}
+</code></pre>
+
+
+
+</details>
+
 <a id="0x1_dkg_initialize"></a>
 
 ## Function `initialize`
diff --git a/aptos-move/framework/aptos-framework/sources/dkg.move b/aptos-move/framework/aptos-framework/sources/dkg.move
index 8e55ccb2ae8fd..92a45f67ceb93 100644
--- a/aptos-move/framework/aptos-framework/sources/dkg.move
+++ b/aptos-move/framework/aptos-framework/sources/dkg.move
@@ -22,6 +22,15 @@ module aptos_framework::dkg {
         target_validator_set: vector<ValidatorConsensusInfo>,
     }
 
+    enum TestEnum has drop {
+        A,
+        B
+    }
+
+    fun test_a(): TestEnum {
+       TestEnum::A
+    }
+
     #[event]
     struct DKGStartEvent has drop, store {
         session_metadata: DKGSessionMetadata,
@@ -118,4 +127,10 @@ module aptos_framework::dkg {
     public fun session_dealer_epoch(session: &DKGSessionState): u64 {
         session.metadata.dealer_epoch
     }
+
+    #[test]
+    public entry fun test() {
+        test_a() == TestEnum::A;
+    }
+
 }