Actually use the next[] shape array in default AI kernel
Fix NPE