fix k / kg bug