diff --git a/config_test.go b/config_test.go
index 6a65fe5b..2985b234 100644
--- a/config_test.go
+++ b/config_test.go
@@ -26,7 +26,7 @@ func TestLoadConfig(t *testing.T) {
 
 func TestLoadConfigNotFound(t *testing.T) {
 	opt, err := LoadConfig("nonexistent.yaml")
-	assert.Error(t, err)
+	require.Error(t, err)
 	assert.Nil(t, opt)
 }
 
@@ -41,7 +41,7 @@ func TestLoadConfigInvalidYAML(t *testing.T) {
 	tmpFile.Close()
 
 	opt, err := LoadConfig(tmpFile.Name())
-	assert.Error(t, err)
+	require.Error(t, err)
 	assert.Nil(t, opt)
 }
 
diff --git a/listener_test.go b/listener_test.go
index 32ce7026..631bbaad 100644
--- a/listener_test.go
+++ b/listener_test.go
@@ -71,7 +71,7 @@ func TestLimitedListenerAtLimit(t *testing.T) {
 	client := &http.Client{Timeout: 500 * time.Millisecond}
 	_, err := client.Get(server.URL)
 	// Connection should be rejected
-	assert.Error(t, err, "expected connection to be rejected")
+	require.Error(t, err, "expected connection to be rejected")
 
 	// Release the blocked connections
 	close(block)
@@ -106,7 +106,7 @@ func TestLimitedConnRelease(t *testing.T) {
 	// Second request should fail
 	client := &http.Client{Timeout: 200 * time.Millisecond}
 	_, err := client.Get(server.URL)
-	assert.Error(t, err, "expected connection to be rejected when limit reached")
+	require.Error(t, err, "expected connection to be rejected when limit reached")
 
 	// Release the first connection
 	close(block)