diff --git a/internal/agent/agent_test.go b/internal/agent/agent_test.go
index 3980a965321c7615594a4d413e0d0f1c39b256cb..906b8df89073d9faab428851a7e76e27644702e5 100644
--- a/internal/agent/agent_test.go
+++ b/internal/agent/agent_test.go
@@ -12,6 +12,7 @@ import (
 	"github.com/charmbracelet/crush/internal/session"
 	"github.com/charmbracelet/fantasy/ai"
 	"github.com/charmbracelet/fantasy/anthropic"
+	"github.com/stretchr/testify/assert"
 	"github.com/stretchr/testify/require"
 	"gopkg.in/dnaeon/go-vcr.v4/pkg/recorder"
 
@@ -23,12 +24,12 @@ type builderFunc func(r *recorder.Recorder) (ai.LanguageModel, error)
 func TestSessionSimpleAgent(t *testing.T) {
 	r := newRecorder(t)
 	sonnet, err := anthropicBuilder("claude-sonnet-4-5-20250929")(r)
-	require.Nil(t, err)
+	require.NoError(t, err)
 	haiku, err := anthropicBuilder("claude-3-5-haiku-20241022")(r)
-	require.Nil(t, err)
+	require.NoError(t, err)
 	agent, sessions, messages := testSessionAgent(t, sonnet, haiku, "You are a helpful assistant")
 	session, err := sessions.Create(t.Context(), "New Session")
-	require.Nil(t, err)
+	require.NoError(t, err)
 
 	res, err := agent.Run(t.Context(), SessionAgentCall{
 		Prompt:          "Hello",
@@ -36,14 +37,14 @@ func TestSessionSimpleAgent(t *testing.T) {
 		MaxOutputTokens: 10000,
 	})
 
-	require.Nil(t, err)
-	require.NotNil(t, res)
+	require.NoError(t, err)
+	assert.NotNil(t, res)
 
 	t.Run("should create session messages", func(t *testing.T) {
 		msgs, err := messages.List(t.Context(), session.ID)
-		require.Nil(t, err)
+		require.NoError(t, err)
 		// Should have the agent and user message
-		require.Equal(t, len(msgs), 2)
+		assert.Equal(t, len(msgs), 2)
 	})
 }