fixing test of neural network