Bug fixes
0.2.2