Fix KT infinity bug