Add forgotten arguments

(cherry picked from commit 6184ea0b0edfe0bd37e9d6efe597cdda28ab4f96)