diff --git a/tools/demo.py b/tools/demo.py
index 631c68a..522c3ce 100755
--- a/tools/demo.py
+++ b/tools/demo.py
@@ -128,6 +128,7 @@ def parse_args():
 
     if args.cpu_mode:
         caffe.set_mode_cpu()
+        cfg.USE_GPU_NMS = False
     else:
         caffe.set_mode_gpu()
         caffe.set_device(args.gpu_id)
diff --git a/tools/demo_rfcn.py b/tools/demo_rfcn.py
index 1cc6083..41556e9 100755
--- a/tools/demo_rfcn.py
+++ b/tools/demo_rfcn.py
@@ -127,6 +127,7 @@ def parse_args():
 
     if args.cpu_mode:
         caffe.set_mode_cpu()
+        cfg.USE_GPU_NMS = False
     else:
         caffe.set_mode_gpu()
         caffe.set_device(args.gpu_id)