diff --git a/runtime/v2/shim.go b/runtime/v2/shim.go
index 982d1bb34..c0d3833c5 100644
--- a/runtime/v2/shim.go
+++ b/runtime/v2/shim.go
@@ -112,7 +112,7 @@ func (s *shim) Shutdown(ctx context.Context) error {
 	_, err := s.task.Shutdown(ctx, &task.ShutdownRequest{
 		ID: s.ID(),
 	})
-	if err != nil && err != ttrpc.ErrClosed {
+	if err != nil && errors.Cause(err) != ttrpc.ErrClosed {
 		return errdefs.FromGRPC(err)
 	}
 	return nil