Small fix
1.0.1