minor bug fix 2