Fix unit tests adding a forgotten arg.