1 files changed, 8 insertions, 3 deletions
diff --git a/src/test/java/com/google/devtools/build/lib/skylarkinterface/processor/testsources/GoldenCase.java b/src/test/java/com/google/devtools/build/lib/skylarkinterface/processor/testsources/GoldenCase.java
index 815ba22bf1..1eb207719f 100644
--- a/src/test/java/com/google/devtools/build/lib/skylarkinterface/processor/testsources/GoldenCase.java
+++ b/src/test/java/com/google/devtools/build/lib/skylarkinterface/processor/testsources/GoldenCase.java
@@ -16,6 +16,7 @@ package com.google.devtools.build.lib.skylarkinterface.processor.testsources;
 
 import com.google.devtools.build.lib.events.Location;
 import com.google.devtools.build.lib.skylarkinterface.Param;
+import com.google.devtools.build.lib.skylarkinterface.ParamType;
 import com.google.devtools.build.lib.skylarkinterface.SkylarkCallable;
 import com.google.devtools.build.lib.syntax.Environment;
 import com.google.devtools.build.lib.syntax.FuncallExpression;
@@ -74,10 +75,14 @@ public class GoldenCase {
     parameters = {
       @Param(name = "one", type = String.class, named = true),
       @Param(name = "two", type = Integer.class, named = true),
-      @Param(name = "three", type = String.class, named = true, defaultValue = "None",
-          noneable = true),
+      @Param(name = "three",
+          allowedTypes = {
+              @ParamType(type = String.class),
+              @ParamType(type = Integer.class),
+          },
+          named = true, defaultValue = "None", noneable = true),
     })
-  public String threeArgMethodWithParams(String one, Integer two, String three) {
+  public String threeArgMethodWithParams(String one, Integer two, Object three) {
     return "baz";
   }