diff --git a/ql/test/library-tests/semmle/go/frameworks/StdlibTaintFlow/main.go b/ql/test/library-tests/semmle/go/frameworks/StdlibTaintFlow/main.go
index 197f9809576..110a1881b4b 100644
--- a/ql/test/library-tests/semmle/go/frameworks/StdlibTaintFlow/main.go
+++ b/ql/test/library-tests/semmle/go/frameworks/StdlibTaintFlow/main.go
@@ -1,10 +1,10 @@
 package main
 
 func main()              {}
-func sink(v interface{}) {}
+func sink(id int, v interface{}) {}
 
 func link(from interface{}, into interface{}) {}
 
-func newSource() interface{} {
+func newSource(id int) interface{} {
 	return nil
 }