correct minor bug
1.0.4