Bug fixes (v0.2.2.1)