small bug fix
0.6.1