diff --git a/packages/@aws-cdk/aws-cloudfront/lib/experimental/edge-function.ts b/packages/@aws-cdk/aws-cloudfront/lib/experimental/edge-function.ts
index 2d62fe984f9b9..7a04feb1fb328 100644
--- a/packages/@aws-cdk/aws-cloudfront/lib/experimental/edge-function.ts
+++ b/packages/@aws-cdk/aws-cloudfront/lib/experimental/edge-function.ts
@@ -201,8 +201,8 @@ export class EdgeFunction extends Resource implements lambda.IVersion {
   }
 
   private edgeStack(stackId?: string): Stack {
-    const stage = this.node.root;
-    if (!stage || !Stage.isStage(stage)) {
+    const stage = Stage.of(this);
+    if (!stage) {
       throw new Error('stacks which use EdgeFunctions must be part of a CDK app or stage');
     }
     const region = this.env.region;
diff --git a/packages/@aws-cdk/aws-cloudfront/test/experimental/edge-function.test.ts b/packages/@aws-cdk/aws-cloudfront/test/experimental/edge-function.test.ts
index b11ab1bbeb4dd..8ec045cfb2ae1 100644
--- a/packages/@aws-cdk/aws-cloudfront/test/experimental/edge-function.test.ts
+++ b/packages/@aws-cdk/aws-cloudfront/test/experimental/edge-function.test.ts
@@ -158,6 +158,32 @@ describe('stacks', () => {
     const fn2Stack = app.node.findChild(fn2StackId) as cdk.Stack;
     expect(fn2Stack).toCountResources('AWS::Lambda::Function', 1);
   });
+
+  test('cross-region stack supports defining functions within stages', () => {
+    app = new cdk.App();
+    const stage = new cdk.Stage(app, 'Stage');
+    stack = new cdk.Stack(stage, 'Stack', {
+      env: { account: '111111111111', region: 'testregion' },
+    });
+
+    new cloudfront.experimental.EdgeFunction(stack, 'MyFn', defaultEdgeFunctionProps());
+
+    // Because 'expect(stack)' doesn't work correctly for stacks in nested assemblies
+    const stackArtifact = stage.synth().getStackArtifact(stack.artifactId);
+    expect(stackArtifact).toHaveResourceLike('AWS::Lambda::Function', {
+      Handler: '__entrypoint__.handler',
+      Role: {
+        'Fn::GetAtt': ['CustomCrossRegionStringParameterReaderCustomResourceProviderRole71CD6825', 'Arn'],
+      },
+    });
+    expect(stackArtifact).toHaveResource('Custom::CrossRegionStringParameterReader', {
+      ServiceToken: {
+        'Fn::GetAtt': ['CustomCrossRegionStringParameterReaderCustomResourceProviderHandler65B5F33A', 'Arn'],
+      },
+      Region: 'us-east-1',
+      ParameterName: 'EdgeFunctionArnMyFn',
+    });
+  });
 });
 
 test('addAlias() creates alias in function stack', () => {