KI fixes