bug fix on input parameter