diff --git a/lazy.go b/lazy.go
index 79cb897..b4ed27a 100644
--- a/lazy.go
+++ b/lazy.go
@@ -49,7 +49,7 @@ func GenerateRandomResults(query string, exampleObj any, keyVal []any, rowCount
 			}
 
 			if field.Tag.Get("test") == "key" {
-				rows[y] = append(rows[y], keyVal)
+				rows[y] = append(rows[y], keyVal[y])
 				continue
 			}