minor bug fix #2