Modified test for CNNs