diff --git a/deploy/TensorRT/cpp/include/bytetrack/BYTETracker.h b/deploy/TensorRT/cpp/include/bytetrack/BYTETracker.h
index c9bb696a..06aaf416 100644
--- a/deploy/TensorRT/cpp/include/bytetrack/BYTETracker.h
+++ b/deploy/TensorRT/cpp/include/bytetrack/BYTETracker.h
@@ -2,6 +2,15 @@
 
 #include "STrack.h"
 
+// struct Object
+// {
+//     cv::Rect_<float> rect;
+//     int label;
+//     float prob;
+// };
+
+namespace bytetrack {
+
 struct Object
 {
     cv::Rect_<float> rect;
@@ -9,8 +18,6 @@ struct Object
     float prob;
 };
 
-namespace bytetrack {
-
 class BYTETracker
 {
 public:
diff --git a/deploy/TensorRT/cpp/include/bytetrack/STrack.h b/deploy/TensorRT/cpp/include/bytetrack/STrack.h
index 4bd9d000..3c76cd16 100644
--- a/deploy/TensorRT/cpp/include/bytetrack/STrack.h
+++ b/deploy/TensorRT/cpp/include/bytetrack/STrack.h
@@ -6,10 +6,11 @@
 using namespace cv;
 using namespace std;
 
+namespace bytetrack {
+
 enum TrackState { New = 0, Tracked, Lost, Removed };
 
-class STrack
-{
+class STrack {
 public:
 	STrack(vector<float> tlwh_, float score);
 	~STrack();
@@ -47,4 +48,6 @@ class STrack
 
 private:
 	byte_kalman::KalmanFilter kalman_filter;
-};
\ No newline at end of file
+};
+
+}  // namespace bytetrack
\ No newline at end of file
diff --git a/deploy/TensorRT/cpp/src/STrack.cpp b/deploy/TensorRT/cpp/src/STrack.cpp
index ceb08c7c..78b6575d 100644
--- a/deploy/TensorRT/cpp/src/STrack.cpp
+++ b/deploy/TensorRT/cpp/src/STrack.cpp
@@ -1,5 +1,7 @@
 #include "STrack.h"
 
+namespace bytetrack {
+
 STrack::STrack(vector<float> tlwh_, float score)
 {
 	_tlwh.resize(4);
@@ -189,4 +191,6 @@ void STrack::multi_predict(vector<STrack*> &stracks, byte_kalman::KalmanFilter &
 		}
 		kalman_filter.predict(stracks[i]->mean, stracks[i]->covariance);
 	}
-}
\ No newline at end of file
+}
+
+}  // namespace bytetrack
\ No newline at end of file