diff --git a/Makefile b/Makefile
index 04ad62d6..1eb08f5b 100644
--- a/Makefile
+++ b/Makefile
@@ -25,7 +25,7 @@ unexport NO_OPENMP
 # all tge possible warning tread
 WARNFLAGS= -Wall
 CFLAGS = -DMSHADOW_FORCE_STREAM $(WARNFLAGS)
-CFLAGS += -g -O3 -I./mshadow/ -I./dmlc-core/include -fPIC $(MSHADOW_CFLAGS) $(DMLC_CFLAGS)
+CFLAGS += -g -O3 -I./mshadow/ -I./dmlc-core/include -I./ps-lite/deps/include -fPIC $(MSHADOW_CFLAGS) $(DMLC_CFLAGS)
 LDFLAGS = -pthread $(MSHADOW_LDFLAGS) $(DMLC_LDFLAGS)
 NVCCFLAGS = --use_fast_math -g -O3 -ccbin $(CXX) $(MSHADOW_NVCCFLAGS)
 ROOTDIR = $(CURDIR)
diff --git a/wrapper/cxxnet_wrapper.cpp b/wrapper/cxxnet_wrapper.cpp
index bfa9c8f8..0a2630d2 100644
--- a/wrapper/cxxnet_wrapper.cpp
+++ b/wrapper/cxxnet_wrapper.cpp
@@ -81,7 +81,6 @@ class WrapperNet {
   WrapperNet(const char *device, const char *s_cfg)
       : res_pred(false), temp2(false),
         temp4(false), net_(NULL) {
-    device = "gpu";
     net_type = 0;
     silent = 0;
     print_step = 100;