diff --git a/src/ai/evaluations.py b/src/ai/evaluations.py
index 4f739fb..e8ed37d 100644
--- a/src/ai/evaluations.py
+++ b/src/ai/evaluations.py
@@ -4,7 +4,8 @@ import time
 import neat
 import numpy as np
 import pygame
-from game import Block, Main
+from game import Main
+from game.sprites import Block
 from utils import CONFIG
 
 from .fitness import calculate_fitness
diff --git a/src/ai/fitness/wells.py b/src/ai/fitness/wells.py
index 06acebc..1ffd872 100644
--- a/src/ai/fitness/wells.py
+++ b/src/ai/fitness/wells.py
@@ -2,8 +2,6 @@ from typing import Optional
 
 import numpy as np
 
-from ai.log import log
-
 from .peaks import get_peaks