diff --git a/internal/tests/prediction_test.go b/internal/tests/prediction_test.go
index 015314a..77e4c08 100644
--- a/internal/tests/prediction_test.go
+++ b/internal/tests/prediction_test.go
@@ -7,6 +7,7 @@ import (
 	"io"
 	"net/http"
 	"testing"
+	"time"
 
 	"github.com/replicate/go/must"
 
@@ -97,6 +98,7 @@ func TestPredictionCrash(t *testing.T) {
 		assert.Equal(t, http.StatusInternalServerError, resp.StatusCode)
 		body := string(must.Get(io.ReadAll(resp.Body)))
 		assert.Equal(t, "Internal Server Error", body)
+		time.Sleep(time.Second)
 		assert.Equal(t, "DEFUNCT", ct.HealthCheck().Status)
 	} else {
 		resp := ct.Prediction(map[string]any{"i": 1, "s": "bar"})