diff --git a/agbenchmark/generate_test.py b/agbenchmark/generate_test.py
index 8eaeab96f..fa655b569 100644
--- a/agbenchmark/generate_test.py
+++ b/agbenchmark/generate_test.py
@@ -134,7 +134,7 @@ def create_single_test(
 
         scores = self.get_scores(config)
         request.node.scores = scores  # store scores in request.node
-        assert 1 in scores
+        assert 1 in scores["values"]
 
     # Parametrize the method here
     test_method = pytest.mark.parametrize(