Fix bug and bump version to 1.0.1