bug in mnist_cnn