diff --git a/src/view/gui.py b/src/view/gui.py
index e42bb52..da3849a 100644
--- a/src/view/gui.py
+++ b/src/view/gui.py
@@ -1,7 +1,7 @@
 import tkinter as tk
 
 from logic.board import Board
-from logic.pieces.piece import Piece
+from logic.pieces.piece import Colour, Piece
 from logic.position import Position
 from view.view import View