fix for convolutional networks - it looks like they are working now