Merge pull request #3 from avsm/master

fix Lwt_fiber fast path
jaked · Jun 17, 2011 · 48116cf · 48116cf
2 parents 8dbab94 + 9a656ba
commit 48116cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/lwt-fiber/lwt_fiber.ml b/src/lwt-fiber/lwt_fiber.ml
@@ -22,11 +22,11 @@ let await t =
     match !active_prompt with
       | None -> failwith "await called outside start"
       | Some p -> p in
-  active_prompt := None;
 
   match Lwt.poll t with
     | Some v -> v
     | None ->
+        active_prompt := None;
         Delimcc.shift0 p begin fun k ->
           let ready _ =
             active_prompt := Some p;