More NTK tests