diff --git a/runtime/v2/shim.go b/runtime/v2/shim.go
index c652d3af5..a8efdbce3 100644
--- a/runtime/v2/shim.go
+++ b/runtime/v2/shim.go
@@ -100,6 +100,8 @@ func loadShim(ctx context.Context, bundle *Bundle, events *exchange.Exchange, rt
 		events:  events,
 		rtTasks: rt,
 	}
+	ctx, cancel := context.WithTimeout(ctx, 5*time.Second)
+	defer cancel()
 	if err := s.Connect(ctx); err != nil {
 		return nil, err
 	}