minor fix in NN usage