diff --git a/main.py b/main.py
index b67cb2c8b59aaf6669bb2714a5ab4ac169a5b37e..139bbe9a25bbf79503f387442df7ea960862a870 100644
--- a/main.py
+++ b/main.py
@@ -1,3 +1,4 @@
+from matplotlib import pyplot as plt
 from cubeplot import draw_cube, draw_face
 import numpy as np
 from ultralytics import YOLO