diff --git a/tests/loss_function_test.py b/tests/loss_function_test.py
index 637d6f3..fbc982c 100644
--- a/tests/loss_function_test.py
+++ b/tests/loss_function_test.py
@@ -2,7 +2,7 @@ from asyncio import base_tasks
 import torch
 import pytest
 
-from enhancer.utils.loss import mean_absolute_error, mean_squared_error
+from enhancer.loss import mean_absolute_error, mean_squared_error
 
 loss_functions = [mean_absolute_error(), mean_squared_error()]