Small bug fix
upgrade to 0.1.6