Fix missing parameters from refactoring the code.