diff --git a/internal/container/container.go b/internal/container/container.go
index 698230e..0e240a7 100644
--- a/internal/container/container.go
+++ b/internal/container/container.go
@@ -243,6 +243,7 @@ func (a *Client) RunContainer(ctx context.Context, params RunContainerParams) (<
 
 		a.runContainerLoop(
 			ctx,
+			cancel,
 			createResp.ID,
 			params.ContainerConfig.Image,
 			params.NetworkCountConfig,
@@ -329,6 +330,7 @@ func (a *Client) pullImageIfNeeded(ctx context.Context, imageName string, contai
 // when the container exits.
 func (a *Client) runContainerLoop(
 	ctx context.Context,
+	cancel context.CancelFunc,
 	containerID string,
 	imageName string,
 	networkCountConfig NetworkCountConfig,
@@ -336,6 +338,8 @@ func (a *Client) runContainerLoop(
 	stateC chan<- domain.Container,
 	errC chan<- error,
 ) {
+	defer cancel()
+
 	type containerWaitResponse struct {
 		container.WaitResponse