From a98669bb990cd77e66910cd66e0c2a27cd6013f9 Mon Sep 17 00:00:00 2001
From: Yoshua Wuyts <yoshuawuyts@gmail.com>
Date: Sun, 3 Nov 2019 00:38:50 +0100
Subject: [PATCH] revert #7

Signed-off-by: Yoshua Wuyts <yoshuawuyts@gmail.com>
---
 src/lib.rs | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index 77ecd8a..2450bd5 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -147,9 +147,7 @@ pub fn test(_attr: TokenStream, item: TokenStream) -> TokenStream {
 ///
 /// #[async_attributes::bench]
 /// async fn bench_1(b: &mut test::Bencher) {
-///     b.iter(|| {
-///         println!("hello world");
-///     })
+///     println!("hello world");
 /// }
 /// ```
 #[proc_macro_attribute]
@@ -178,9 +176,9 @@ pub fn bench(_attr: TokenStream, item: TokenStream) -> TokenStream {
         #[bench]
         #(#attrs)*
         fn #name(b: &mut test::Bencher) #ret {
-            task::block_on(task::spawn(async {
-                #body
-            }))
+            b.iter(|| {
+                let _ = async_std::task::block_on(async { #body });
+            });
         }
     };