diff --git a/lab4/main.py b/lab4/main.py
index b0ea64c..18e8326 100644
--- a/lab4/main.py
+++ b/lab4/main.py
@@ -26,9 +26,9 @@ from gp.mutations import (
     ShrinkMutation,
 )
 from gp.ops import ADD, COS, DIV, EXP, MUL, POW, SIN, SQUARE, SUB
-from gp.population import ramped_initialization
 from gp.primitive import Var
 from gp.selection import tournament_selection
+from gp.utils import ramped_initialization
 
 NUM_VARS = 8
 TEST_POINTS = 10000