diff --git a/flake_test.go b/flake_test.go
index 08c00fb..a6b86d7 100644
--- a/flake_test.go
+++ b/flake_test.go
@@ -10,7 +10,7 @@ func BenchmarkGenerate(b *testing.B) {
 
 	b.ReportAllocs()
 	for n := 0; n < b.N; n++ {
-		_, _ = node.GenerateNoSleep()
+		_, _ = node.Generate()
 	}
 
 }