fixed a typo in init of NN