small change to fix parameters on some functions