Small fix 0.0.2