Adjust to correct variable in params class