Fixed up mnist_nn_tests