diff --git a/app/src/main/java/school_project/GameUI.java b/app/src/main/java/school_project/GameUI.java
index 82b4cf8..d15a53b 100644
--- a/app/src/main/java/school_project/GameUI.java
+++ b/app/src/main/java/school_project/GameUI.java
@@ -19,7 +19,7 @@ public class GameUI extends BorderPane {
         pieces.setSpacing(10);
         for (Piece p : level.getPieces()) {
             MatrixShape _piece = new MatrixShape(p);
-            _piece.setColor(Color.RED); // TODO: Change with piece color
+            _piece.setColor(p.getColor());
             pieces.getChildren().add(_piece);
         }