fix to do convolutional predictions