NTK fixes