diff --git a/face_swapper/src/training.py b/face_swapper/src/training.py
index 182fb58..c2d512b 100644
--- a/face_swapper/src/training.py
+++ b/face_swapper/src/training.py
@@ -1,7 +1,6 @@
 import configparser
 import random
-
-from sympy.stats.sampling.sample_numpy import numpy
+import numpy
 
 from typing import Tuple
 import os