diff --git a/policy/src/main/java/dev/cel/policy/CelPolicyConfig.java b/policy/src/main/java/dev/cel/policy/CelPolicyConfig.java
index c5acc43f..d3786d51 100644
--- a/policy/src/main/java/dev/cel/policy/CelPolicyConfig.java
+++ b/policy/src/main/java/dev/cel/policy/CelPolicyConfig.java
@@ -127,6 +127,10 @@ public Cel extend(Cel cel, CelOptions celOptions) throws CelPolicyValidationExce
                       .map(f -> f.toCelFunctionDecl(celTypeProvider))
                       .collect(toImmutableList()));
 
+      if (!container().isEmpty()) {
+        celBuilder.setContainer(container());
+      }
+
       addAllExtensions(celBuilder, celOptions);
 
       return celBuilder.build();