Bug fixes
0.4.1