diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
index 2d9c1c39..8566f1c8 100644
--- a/Dockerfile.tmpl
+++ b/Dockerfile.tmpl
@@ -112,7 +112,7 @@ RUN conda config --add channels nvidia && \
 # b/341938540: unistall grpc-cpp to allow >=v24.4 cudf and cuml to be installed.
 {{ if eq .Accelerator "gpu" }}
 RUN pip uninstall -y pyarrow && \
-    mamba remove grpc-cpp && \
+    mamba remove --force grpc-cpp && \
     mamba install -y -c conda-forge spacy cudf>=24.4 cuml>=24.4 cupy cuda-version=$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION && \
     /tmp/clean-layer.sh
 {{ else }}
diff --git a/tests/common.py b/tests/common.py
index 175a6206..30a7bb0f 100644
--- a/tests/common.py
+++ b/tests/common.py
@@ -7,7 +7,7 @@
 def getAcceleratorName():
     try:
         deviceName = subprocess.check_output(['nvidia-smi', '--query-gpu=name', '--format=csv,noheader'])
-        return deviceName.decode('utf-8').strip();
+        return deviceName.decode('utf-8').strip()
     except FileNotFoundError:
         return("nvidia-smi not found.")