Bug fixes (v0.4.1)