diff --git a/pipe_test.go b/pipe_test.go
index d69ceb4..964cadc 100644
--- a/pipe_test.go
+++ b/pipe_test.go
@@ -53,8 +53,8 @@ func TestDialListenerGetsCancelled(t *testing.T) {
 	time.Sleep(time.Millisecond * 30)
 	cancel()
 	err = <-ch
-	if err == nil {
-		t.Fatalf("expected ErrTimeout, got %v", err)
+	if err != context.Canceled {
+		t.Fatalf("expected context.Canceled, got %v", err)
 	}
 }