diff --git a/internal/container/container_test.go b/internal/container/container_test.go
index 4dc8df6..654fea7 100644
--- a/internal/container/container_test.go
+++ b/internal/container/container_test.go
@@ -221,7 +221,7 @@ func TestClientRunContainerWithRestart(t *testing.T) {
 	assert.Equal(t, "restarting", state.Status)
 	assert.Equal(t, "unhealthy", state.HealthState)
 	assert.Nil(t, state.ExitCode)
-	assert.Equal(t, 0, state.RestartCount) // not incremented until the actual restart
+	assert.Zero(t, state.RestartCount) // not incremented until the actual restart
 
 	// During the restart, the "running" status is triggered by Docker events
 	// only. So we don't expect one in unit tests. (Probably the initial startup