diff --git a/.librarian/generator-input/client-post-processing/bigquery-storage-integration.yaml b/.librarian/generator-input/client-post-processing/bigquery-storage-integration.yaml
index 0a495b68d521..6449c88e1791 100644
--- a/.librarian/generator-input/client-post-processing/bigquery-storage-integration.yaml
+++ b/.librarian/generator-input/client-post-processing/bigquery-storage-integration.yaml
@@ -28,6 +28,15 @@ replacements:
           "pyarrow",
       ]
     count: 1
+  - paths: [
+      packages/google-cloud-bigquery-storage/noxfile.py,
+    ]
+    before: |
+      \    # Install all dependencies\n    session.install\("-e", "\."\)
+    after: |
+          # Install all dependencies
+          session.install("-e", f".[{','.join(UNIT_TEST_EXTRAS)}]")
+    count: 2
   - paths: [
       packages/google-cloud-bigquery-storage/noxfile.py,
     ]
diff --git a/packages/google-cloud-bigquery-storage/noxfile.py b/packages/google-cloud-bigquery-storage/noxfile.py
index e7c22b165e90..4ec53338e333 100644
--- a/packages/google-cloud-bigquery-storage/noxfile.py
+++ b/packages/google-cloud-bigquery-storage/noxfile.py
@@ -507,7 +507,7 @@ def prerelease_deps(session, protobuf_implementation):
     """
 
     # Install all dependencies
-    session.install("-e", ".")
+    session.install("-e", f".[{','.join(UNIT_TEST_EXTRAS)}]")
 
     # Install dependencies for the unit test environment
     unit_deps_all = UNIT_TEST_STANDARD_DEPENDENCIES + UNIT_TEST_EXTERNAL_DEPENDENCIES
@@ -618,7 +618,7 @@ def core_deps_from_source(session, protobuf_implementation):
     """
 
     # Install all dependencies
-    session.install("-e", ".")
+    session.install("-e", f".[{','.join(UNIT_TEST_EXTRAS)}]")
 
     # Install dependencies for the unit test environment
     unit_deps_all = UNIT_TEST_STANDARD_DEPENDENCIES + UNIT_TEST_EXTERNAL_DEPENDENCIES