diff --git a/testing/internal/e2e/tests/base/target_tcp_connect_mysql_test.go b/testing/internal/e2e/tests/base/target_tcp_connect_mysql_test.go
index f6331807cd..64648a4e1b 100644
--- a/testing/internal/e2e/tests/base/target_tcp_connect_mysql_test.go
+++ b/testing/internal/e2e/tests/base/target_tcp_connect_mysql_test.go
@@ -117,7 +117,14 @@ func TestCliTcpTargetConnectMysql(t *testing.T) {
 
 	var buf bytes.Buffer
 	_, _ = io.Copy(&buf, f)
-	require.Contains(t, buf.String(), "mysql>", "Session did not return expected MySQL prompt")
-	require.Contains(t, buf.String(), mysqlDbName, "Session did not return expected output")
+	
+	output := buf.String()
+	t.Logf("MySQL session output: %s", output)
+	
+	require.Contains(t, output, "| "+mysqlDbName+" |", "Session did not return expected database query result")
+	
+	require.True(t, 
+		strings.Contains(output, "mysql>") || strings.Contains(output, "MySQL ["),
+		"Session did not show MySQL/MariaDB prompt")
 	t.Log("Successfully connected to MySQL target")
 }