Improve chess game editor

1 files changed, 3 insertions, 4 deletions

diff --git a/src/game_tab/right_panel/editor/EditorCanvas.cpp b/src/game_tab/right_panel/editor/EditorCanvas.cpp
index c4d389d..0a32589 100644
--- a/src/game_tab/right_panel/editor/EditorCanvas.cpp
+++ b/src/game_tab/right_panel/editor/EditorCanvas.cpp
@@ -44,11 +44,13 @@ void EditorCanvas::DrawElement(const cgeditor::Element &e) {
   dc->SetFont(default_font);
   if (e.prop & cgeditor::Property::Rectangle) {
     if (e.prop & cgeditor::Property::Scrollbarbg) {
-      dc->SetBrush(*wxCYAN_BRUSH);
+      dc->SetBrush(wxColour(243,243,243));
     } else if (e.prop & cgeditor::Property::Scrollbar) {
       dc->SetBrush(*wxGREY_BRUSH);
     } else if (e.prop & cgeditor::Property::Margin) {
       dc->SetBrush(wxBrush(wxColour(243,243,243)));
+    } else if (e.prop & cgeditor::Property::Comment) {
+      dc->SetBrush(wxBrush(wxColour(255, 255, 204)));
     } else if (e.prop & cgeditor::Property::Button) {
       if (e.prop & cgeditor::Property::On) {
         dc->DrawBitmap(hide_icon, e.x, e.y);
@@ -85,9 +87,6 @@ void EditorCanvas::DrawElement(const cgeditor::Element &e) {
       }
       dc->DrawBitmap(*t.Get(p), e.x, y);
     } else if (e.prop & cgeditor::Property::Comment) {
-      wxRect recToDraw(e.x, e.y, e.width, e.height);
-      dc->SetBrush(wxBrush(wxColour(255, 255, 204)));
-      dc->DrawRectangle(recToDraw);
       dc->DrawText(wxString(e.text), wxPoint(e.x, e.y));
     } else if (e.prop & cgeditor::Property::Menuitem) {
       wxRect recToDraw(e.x, e.y, e.width, e.height);