diff --git a/include/publisher_vision.h b/include/publisher_vision.h
index 5bcb9339b1753b776822e238f9fa3bf3da54e545..6b00b2fba15e94bb6ae8e24399f300aada6f3491 100644
--- a/include/publisher_vision.h
+++ b/include/publisher_vision.h
@@ -83,7 +83,7 @@ class PublisherVisionDebug : public Publisher
         int size_feature_kfs_pix_;
         double thickness_lmk_;
         double size_text_ID_;
-        int color_tracks_features_;
+        COLORFEATURES color_tracks_features_;
 
         image_transport::Publisher publisher_image_;
         image_transport::ImageTransport img_transport_;
@@ -99,7 +99,7 @@ class PublisherVisionDebug : public Publisher
             Function that return a unique RGB color vector depending on the lenght of the track given,
             so the longer is the track the darker it is
         */
-        std::vector<int> colorTrackAndFeatures(int _nb_feature_in_track, int _max_feature_in_tracks, int _min_feature_in_tracks, int _color_of_features);
+        std::vector<int> colorTrackAndFeatures(int _nb_feature_in_track, int _max_feature_in_tracks, int _min_feature_in_tracks, COLORFEATURES _color_of_features);
 
         /*
             Function that change an image to show tracks and features within it
diff --git a/src/publisher_vision.cpp b/src/publisher_vision.cpp
index d60ebe21622bbf6850898fc35d2ae04bc6eee708..32a8e0b975223559cfa54875e53d0c0c2c362d0d 100644
--- a/src/publisher_vision.cpp
+++ b/src/publisher_vision.cpp
@@ -183,7 +183,7 @@ std::pair<int, int> PublisherVisionDebug::minMaxFeatureInTrack(const std::map<Si
 std::vector<int> PublisherVisionDebug::colorTrackAndFeatures(int _nb_feature_in_track,
                                                              int _max_feature_in_tracks,
                                                              int _min_feature_in_tracks,
-                                                             int _color_of_features)
+                                                             COLORFEATURES _color_of_features)
 {
 
     // std::cout << "\n\ncolorTrackAndFeatures" << '\n';