MNIST fix