fix minor bug
0.1.6