modify a bug of nn