small change to make it runable by caffe!