bug fix in edge case