diff --git a/.github/workflows/ci-test.yml b/.github/workflows/ci-test.yml
index d636e3c1..ca81b9a2 100644
--- a/.github/workflows/ci-test.yml
+++ b/.github/workflows/ci-test.yml
@@ -9,6 +9,9 @@ on:
       - 'docs/**'
   workflow_dispatch:
 
+env:
+  USE_BAZEL_VERSION: 7.7.0
+
 jobs:
   unit-tests:
     if: github.actor != 'copybara-service[bot]'
@@ -31,7 +34,10 @@ jobs:
 
     - name: Install dependencies
       run: |
-        pip install .[test]
+        sudo apt update
+        sudo apt install -y protobuf-compiler
+        pip install --upgrade pip setuptools wheel numpy pyarrow
+        pip install --no-build-isolation .[test]
 
     - name: Run unit tests
       shell: bash
diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml
index 5fd66221..5f1fca26 100644
--- a/.github/workflows/wheels.yml
+++ b/.github/workflows/wheels.yml
@@ -6,6 +6,9 @@ on:
   release:
     types: [published]
 
+env:
+  USE_BAZEL_VERSION: 7.7.0
+
 jobs:
   build-package:
     name: Build sdist
@@ -20,11 +23,15 @@ jobs:
         python-version: '3.10'
 
     - name: Install python dependencies
-      run: pip install --upgrade pip build twine
+      run: |
+        sudo apt update
+        sudo apt install -y protobuf-compiler
+        pip install --upgrade pip setuptools wheel numpy pyarrow build twine
+        pip install --no-build-isolation .
 
     - name: Build sdist and wheel
       run: |
-        python -m build --sdist --wheel -o wheelhouse
+        python -m build --sdist --wheel --no-isolation -o wheelhouse
 
     - name: List and check sdist
       run: |
diff --git a/setup.py b/setup.py
index 28568c48..d9eb3939 100644
--- a/setup.py
+++ b/setup.py
@@ -52,12 +52,12 @@ def _make_required_install_packages():
         "tensorflow>=2.21,<2.22",
         "tensorflow-metadata"
         + select_constraint(
-            default=">=1.17.1,<1.18.0",
+            default="@git+https://github.com/tensorflow/metadata@master",
             nightly=">=1.18.0.dev",
             git_master="@git+https://github.com/tensorflow/metadata@master",
         ),
         "tf_keras>=2",
-        "tfx-bsl@git+https://github.com/tensorflow/tfx-bsl.git@master",
+        "tfx-bsl@git+https://github.com/tensorflow/tfx-bsl@master",
     ]