Fix NKI loading