Another bug fix in doc of stepfunction