diff --git a/morpheus/_lib/doca/include/morpheus/doca/doca_source_stage.hpp b/morpheus/_lib/doca/include/morpheus/doca/doca_source_stage.hpp
index a810a04a83..c14e598734 100644
--- a/morpheus/_lib/doca/include/morpheus/doca/doca_source_stage.hpp
+++ b/morpheus/_lib/doca/include/morpheus/doca/doca_source_stage.hpp
@@ -22,7 +22,9 @@
 #include "morpheus/messages/raw_packet.hpp"
 
 #include <mrc/segment/builder.hpp>
+#include <mrc/segment/object.hpp>  // for Object
 #include <pymrc/node.hpp>
+#include <rxcpp/rx.hpp>
 
 #include <memory>
 #include <string>
diff --git a/morpheus/_lib/doca/src/doca_convert_stage.cpp b/morpheus/_lib/doca/src/doca_convert_stage.cpp
index 5286aeb089..3373951a4b 100644
--- a/morpheus/_lib/doca/src/doca_convert_stage.cpp
+++ b/morpheus/_lib/doca/src/doca_convert_stage.cpp
@@ -47,7 +47,6 @@
 #include <cstdint>
 #include <exception>  // for exception_ptr
 #include <memory>
-#include <ostream>
 #include <string>
 #include <thread>
 #include <utility>