minor bug fix
add parameters