[HK fix bugs]