fixed issue with the neural network builder