Fix neural networks