diff --git a/bundle/src/test/java/dev/cel/bundle/BUILD.bazel b/bundle/src/test/java/dev/cel/bundle/BUILD.bazel
index a97a8e41..74e2b807 100644
--- a/bundle/src/test/java/dev/cel/bundle/BUILD.bazel
+++ b/bundle/src/test/java/dev/cel/bundle/BUILD.bazel
@@ -24,6 +24,7 @@ java_library(
         "//common/resources/testdata/proto3:standalone_global_enum_java_proto",
         "//common/testing",
         "//common/types",
+        "//common/types:cel_proto_message_types",
         "//common/types:cel_proto_types",
         "//common/types:message_type_provider",
         "//common/types:type_providers",
diff --git a/bundle/src/test/java/dev/cel/bundle/CelImplTest.java b/bundle/src/test/java/dev/cel/bundle/CelImplTest.java
index 018d6321..7fcb7de2 100644
--- a/bundle/src/test/java/dev/cel/bundle/CelImplTest.java
+++ b/bundle/src/test/java/dev/cel/bundle/CelImplTest.java
@@ -72,6 +72,7 @@
 import dev.cel.common.ast.CelExpr.CelList;
 import dev.cel.common.testing.RepeatedTestProvider;
 import dev.cel.common.types.CelKind;
+import dev.cel.common.types.CelProtoMessageTypes;
 import dev.cel.common.types.CelProtoTypes;
 import dev.cel.common.types.CelType;
 import dev.cel.common.types.EnumType;
@@ -1155,7 +1156,7 @@ public void program_messageTypeAddedAsVarWithoutDescriptor_throwsHumanReadableEr
     String packageName = CheckedExpr.getDescriptor().getFile().getPackage();
     Cel cel =
         standardCelBuilderWithMacros()
-            .addVar("parsedExprVar", CelProtoTypes.createMessage(ParsedExpr.getDescriptor()))
+            .addVar("parsedExprVar", CelProtoMessageTypes.createMessage(ParsedExpr.getDescriptor()))
             .build();
     CelValidationException exception =
         assertThrows(