Fixes for 0.3.1