Fix forgotten arguments