diff --git a/packages/@aws-cdk/aws-lambda/lib/runtime.ts b/packages/@aws-cdk/aws-lambda/lib/runtime.ts
index 0f26813b8c4a9..bb24fbba9eeae 100644
--- a/packages/@aws-cdk/aws-lambda/lib/runtime.ts
+++ b/packages/@aws-cdk/aws-lambda/lib/runtime.ts
@@ -1,4 +1,4 @@
-import { BundlingDockerImage } from '@aws-cdk/core';
+import { BundlingDockerImage, DockerImage } from '@aws-cdk/core';
 
 export interface LambdaRuntimeProps {
   /**
@@ -218,7 +218,7 @@ export class Runtime {
     this.supportsInlineCode = !!props.supportsInlineCode;
     this.family = family;
     const imageName = props.bundlingDockerImage ?? `amazon/aws-sam-cli-build-image-${name}`;
-    this.bundlingDockerImage = BundlingDockerImage.fromRegistry(imageName);
+    this.bundlingDockerImage = DockerImage.fromRegistry(imageName);
     this.supportsCodeGuruProfiling = props.supportsCodeGuruProfiling ?? false;
 
     Runtime.ALL.push(this);
diff --git a/packages/@aws-cdk/aws-s3-assets/test/integ.assets.bundling.lit.ts b/packages/@aws-cdk/aws-s3-assets/test/integ.assets.bundling.lit.ts
index b8d78aef6f438..a62554bab726a 100644
--- a/packages/@aws-cdk/aws-s3-assets/test/integ.assets.bundling.lit.ts
+++ b/packages/@aws-cdk/aws-s3-assets/test/integ.assets.bundling.lit.ts
@@ -12,7 +12,7 @@ class TestStack extends Stack {
     const asset = new assets.Asset(this, 'BundledAsset', {
       path: path.join(__dirname, 'markdown-asset'), // /asset-input and working directory in the container
       bundling: {
-        image: DockerImage.fromAsset(path.join(__dirname, 'alpine-markdown')), // Build an image
+        image: DockerImage.fromBuild(path.join(__dirname, 'alpine-markdown')), // Build an image
         command: [
           'sh', '-c', `
             markdown index.md > /asset-output/index.html