diff --git a/Dockerfile b/Dockerfile
index a0bc1905..8b68b5cd 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -34,7 +34,7 @@ EOF
 
 
 # Base image
-FROM rapidsai/miniforge-cuda:cuda${CUDA_VER}-base-${LINUX_VER}-py${PYTHON_VER} as base
+FROM rapidsai/miniforge-cuda:cuda${CUDA_VER}-base-${LINUX_VER}-py${PYTHON_VER} AS base
 ARG CUDA_VER
 ARG PYTHON_VER
 
@@ -67,7 +67,7 @@ CMD ["ipython"]
 
 
 # Notebooks image
-FROM base as notebooks
+FROM base AS notebooks
 
 ARG CUDA_VER
 ARG LINUX_DISTRO
diff --git a/context/test_notebooks.py b/context/test_notebooks.py
index 7b3084fb..7f987009 100755
--- a/context/test_notebooks.py
+++ b/context/test_notebooks.py
@@ -70,6 +70,7 @@ def test_notebook(notebook_file, executed_nb_file):
         warnings = []
         outputs = []
 
+
         # use nbconvert to run the notebook natively
         ep = ExecutePreprocessor(timeout=600, kernel_name="python3", allow_errors=True)
         task_init = timeit.default_timer()