fixed a problem with convolutional neural nets