Fix missing KAT when it's enabled