bug fix #4,5