Fixes for 0.2.1