diff --git a/context/withtimeout_test.go b/context/withtimeout_test.go
index 7cddc5f..e6f5669 100644
--- a/context/withtimeout_test.go
+++ b/context/withtimeout_test.go
@@ -11,19 +11,21 @@
 	"golang.org/x/net/context"
 )
 
-var workComplete <-chan int
-
+// This example passes a context with a timeout to tell a blocking function that
+// it should abandon its work after the timeout elapses.
 func ExampleWithTimeout() {
 	// Pass a context with a timeout to tell a blocking function that it
 	// should abandon its work after the timeout elapses.
-	ctx, cancel := context.WithTimeout(context.Background(), 100*time.Millisecond)
+	ctx, cancel := context.WithTimeout(context.Background(), 50*time.Millisecond)
+	defer cancel()
+
 	select {
-	case <-workComplete:
-		cancel()
+	case <-time.After(1 * time.Second):
 		fmt.Println("overslept")
 	case <-ctx.Done():
 		fmt.Println(ctx.Err()) // prints "context deadline exceeded"
 	}
+
 	// Output:
 	// context deadline exceeded
 }