Small bug fix
0.3.2