fix bug
0.6.1