diff --git a/.github/workflows/build_wheel.yml b/.github/workflows/build_wheel.yml
index 6b749b64fd..f68cedaa32 100644
--- a/.github/workflows/build_wheel.yml
+++ b/.github/workflows/build_wheel.yml
@@ -26,11 +26,6 @@ jobs:
           python -m pip install cibuildwheel
 
       - name: Build wheels
-        env:
-          CIBW_BUILD: "cp36-* cp37-* cp38-* cp39-* cp310-*"
-          CIBW_MANYLINUX_X86_64_IMAGE: ghcr.io/deepmodeling/manylinux_2_28_x86_64_tensorflow
-          CIBW_BEFORE_BUILD: pip install tensorflow
-          CIBW_SKIP: "*-win32 *-manylinux_i686 *-musllinux*"
         run: |
           python -m cibuildwheel --output-dir wheelhouse
       - uses: actions/upload-artifact@v2
diff --git a/pyproject.toml b/pyproject.toml
index 8d4bfad3eb..238021e0ac 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -9,3 +9,9 @@ requires = [
     "ninja; platform_system!='Windows'",
 ]
 
+[tool.cibuildwheel]
+test-command = "dp -h"
+test-requires = "tensorflow"
+build = ["cp36-*", "cp37-*", "cp38-*", "cp39-*", "cp310-*"]
+skip = ["*-win32", "*-manylinux_i686", "*-musllinux*"]
+manylinux-x86_64-image = "ghcr.io/deepmodeling/manylinux_2_28_x86_64_tensorflow"