Fixed bug in pycaffe