diff --git a/test.py b/test.py
index 25ce611e..5536eab0 100644
--- a/test.py
+++ b/test.py
@@ -171,7 +171,7 @@ def test(
         map = cocoEval.stats[1]  # update mAP to pycocotools mAP
 
     # Return results
-    return mp, mr, map, mf1, loss
+    return mp, mr, map, mf1, loss / len(dataloader)
 
 
 if __name__ == '__main__':