diff --git a/test_average_calculate.py b/test_average_calculate.py
index 6b67f0b..6bcafae 100644
--- a/test_average_calculate.py
+++ b/test_average_calculate.py
@@ -1,5 +1,5 @@
 import torch
-from DataContainer import DataContainer
+from functions.DataContainer import DataContainer
 import numpy as np
 from tqdm import trange