fixes neural network