add NMI support