diff --git a/tfjs-backend-webgl/src/canvas_util.ts b/tfjs-backend-webgl/src/canvas_util.ts
index dd940899411..c255ce9cd52 100644
--- a/tfjs-backend-webgl/src/canvas_util.ts
+++ b/tfjs-backend-webgl/src/canvas_util.ts
@@ -34,7 +34,13 @@ export function setWebGLContext(
 
 export function getWebGLContext(webGLVersion: number): WebGLRenderingContext {
   if (!(webGLVersion in contexts)) {
-    contexts[webGLVersion] = getWebGLRenderingContext(webGLVersion);
+    const newCtx = getWebGLRenderingContext(webGLVersion);
+    if (newCtx !== null) {
+      contexts[webGLVersion] = newCtx;
+    } else {
+      console.log('Could not get context for WebGL version', webGLVersion);
+      return null;
+    }
   }
   const gl = contexts[webGLVersion];
   if (gl.isContextLost()) {