fix minor bug, add param