upgrade KZ to 0.5.4