diff --git a/.github/workflows/ci-license.yaml b/.github/workflows/ci-license.yaml
index f706a1cd73bfb..e94d1913cacda 100644
--- a/.github/workflows/ci-license.yaml
+++ b/.github/workflows/ci-license.yaml
@@ -86,4 +86,4 @@ jobs:
 
       - name: license check
         if: ${{ steps.check_changes.outputs.docs_only != 'true' }}
-        run: src/check-binary-license ./distribution/server/target/apache-pulsar-*-bin.tar.gz
+        run: src/check-binary-license.sh ./distribution/server/target/apache-pulsar-*-bin.tar.gz
diff --git a/distribution/io/pom.xml b/distribution/io/pom.xml
index da2cbb628096c..b0abad738c67a 100644
--- a/distribution/io/pom.xml
+++ b/distribution/io/pom.xml
@@ -111,7 +111,7 @@
                   <goal>exec</goal>
                 </goals>
                 <configuration>
-                  <executable>${project.basedir}/../../src/pulsar-io-gen</executable>
+                  <executable>${project.basedir}/../../src/pulsar-io-gen.sh</executable>
                   <outputFile>${project.basedir}/target/pulsar-io-gen.output</outputFile>
                   <arguments>
                     <argument>conf</argument>
diff --git a/src/assembly-source-package.xml b/src/assembly-source-package.xml
index 2677299357f3f..676e603787e7d 100644
--- a/src/assembly-source-package.xml
+++ b/src/assembly-source-package.xml
@@ -90,6 +90,7 @@
       <outputDirectory>/src</outputDirectory>
       <includes>
         <include>*.sh</include>
+        <include>*.py</include>
       </includes>
       <fileMode>0755</fileMode>
     </fileSet>
diff --git a/src/check-binary-license b/src/check-binary-license.sh
similarity index 100%
rename from src/check-binary-license
rename to src/check-binary-license.sh
diff --git a/src/pulsar-io-gen b/src/pulsar-io-gen.sh
similarity index 100%
rename from src/pulsar-io-gen
rename to src/pulsar-io-gen.sh