small fix (v0.1.1)