fix bug and bump version to 0.2.1