upgrade KZ to 0.6.1