diff --git a/demo/common/demo_options.cpp b/demo/common/demo_options.cpp
index 7e70b1a..0b94085 100644
--- a/demo/common/demo_options.cpp
+++ b/demo/common/demo_options.cpp
@@ -6,7 +6,6 @@
 #include <fstream>
 #include <iostream>
 #include <limits>
-#include <sstream>
 
 #include "demo/common/dataset_reader.h"
 #include "edgeflow/log.h"
diff --git a/demo/main.cpp b/demo/main.cpp
index 224274f..e76e1e5 100644
--- a/demo/main.cpp
+++ b/demo/main.cpp
@@ -1,10 +1,7 @@
-#include <fstream>
 #include <iostream>
-#include <memory>
 #include <string>
 #include <vector>
 
-#include "demo/common/dataset_reader.h"
 #include "demo/common/demo_options.h"
 #include "demo/common/demo_registry.h"
 #include "edgeflow/operator/interface.h"