diff --git a/src/main/java/edu/ucr/cs/riple/taint/ucrtainting/stubs/securibench.astub b/src/main/java/edu/ucr/cs/riple/taint/ucrtainting/stubs/securibench.astub
index 3c026cd..32f6737 100644
--- a/src/main/java/edu/ucr/cs/riple/taint/ucrtainting/stubs/securibench.astub
+++ b/src/main/java/edu/ucr/cs/riple/taint/ucrtainting/stubs/securibench.astub
@@ -1,3 +1,4 @@
+import edu.ucr.cs.riple.taint.ucrtainting.qual.RTainted;
 import edu.ucr.cs.riple.taint.ucrtainting.qual.RUntainted;
 
 package javax.servlet;
@@ -132,9 +133,17 @@ public class URLEncoder {
 
 package javax.servlet.http;
 public interface HttpServletRequest {
-    public String getHeader(@RUntainted String name);
+    public @RTainted String getHeader(@RUntainted String name);
     public java.util.Enumeration<String> getHeaders(@RUntainted String name);
-    public String getParameter(@RUntainted String name);
+    public @RTainted String getParameter(@RUntainted String name);
+    public String[] getParameterValues(@RUntainted String name);
+}
+
+package javax.servlet;
+public interface ServletRequest {
+    public @RTainted String getHeader(@RUntainted String name);
+    public java.util.Enumeration<String> getHeaders(@RUntainted String name);
+    public @RTainted String getParameter(@RUntainted String name);
     public String[] getParameterValues(@RUntainted String name);
 }