diff --git a/internal/container/container.go b/internal/container/container.go
index e346506..698230e 100644
--- a/internal/container/container.go
+++ b/internal/container/container.go
@@ -332,7 +332,7 @@ func (a *Client) runContainerLoop(
 	containerID string,
 	imageName string,
 	networkCountConfig NetworkCountConfig,
-	shouldRestart ShouldRestartFunc,
+	shouldRestartFunc ShouldRestartFunc,
 	stateC chan<- domain.Container,
 	errC chan<- error,
 ) {
@@ -366,8 +366,8 @@ func (a *Client) runContainerLoop(
 			respC, errC := a.apiClient.ContainerWait(ctx, containerID, container.WaitConditionNextExit)
 			select {
 			case resp := <-respC:
-				if shouldRestart != nil {
-					shouldRestart, err := shouldRestart(resp.StatusCode, restartCount, time.Since(startedWaitingAt))
+				if shouldRestartFunc != nil {
+					shouldRestart, err := shouldRestartFunc(resp.StatusCode, restartCount, time.Since(startedWaitingAt))
 					if shouldRestart && err != nil {
 						panic(fmt.Errorf("shouldRestart must return nil error if restarting, but returned: %w", err))
 					}