Bug fixes for v0.1.2