diff --git a/src/yolo.c b/src/yolo.c
index 319694b4..d62c533f 100644
--- a/src/yolo.c
+++ b/src/yolo.c
@@ -10,7 +10,7 @@
 #endif
 
 char *voc_names[] = {"aeroplane", "bicycle", "bird", "boat", "bottle", "bus", "car", "cat", "chair", "cow", "diningtable", "dog", "horse", "motorbike", "person", "pottedplant", "sheep", "sofa", "train", "tvmonitor"};
-image voc_labels[80];
+image voc_labels[20];
 
 void train_yolo(char *cfgfile, char *weightfile)
 {
diff --git a/src/yolo_kernels.cu b/src/yolo_kernels.cu
index 78fedafb..a8a131c1 100644
--- a/src/yolo_kernels.cu
+++ b/src/yolo_kernels.cu
@@ -21,6 +21,7 @@ extern "C" void convert_yolo_detections(float *predictions, int classes, int num
 extern "C" void draw_yolo(image im, int num, float thresh, box *boxes, float **probs);
 
 extern "C" char *voc_names[];
+extern "C" image *voc_labels;
 
 static float **probs;
 static box *boxes;