Bug fixes
0.5.1