diff --git a/images/pyspark-notebook/Dockerfile b/images/pyspark-notebook/Dockerfile
index 9d969e412b..28d40dc2d0 100644
--- a/images/pyspark-notebook/Dockerfile
+++ b/images/pyspark-notebook/Dockerfile
@@ -74,8 +74,8 @@ USER ${NB_UID}
 # 1. Check out the Spark branch you are on.
 # 2. Find the pandas version in the file spark/dev/infra/Dockerfile.
 RUN mamba install --yes \
-    'grpcio-status=1.56' \
-    'grpcio=1.56' \
+    'grpcio-status' \
+    'grpcio' \
     'pandas=2.0.3' \
     'pyarrow' && \
     mamba clean --all -f -y && \