fix bug in test_convnet.py