Reverted MNIST fix