diff --git a/tests/loss_function_test.py b/tests/loss_function_test.py
index f08ac65..637d6f3 100644
--- a/tests/loss_function_test.py
+++ b/tests/loss_function_test.py
@@ -21,7 +21,7 @@ def test_loss_input_shapes(loss):
     check_loss_shapes_compatibility(loss)
 
 @pytest.mark.parametrize("loss",loss_functions)
-def test_loss_output_shapes(loss):
+def test_loss_output_type(loss):
 
     batch_size = 4
     prediction, target = torch.rand(batch_size,1,1000),torch.rand(batch_size,1,1000)