bug fix 0.6.1