Bug fix
0.2.1