more changes to fix the param change