From 39eae2dfa515d73f7b38b51778e79f2faebdbcd6 Mon Sep 17 00:00:00 2001
From: Mayur Singal <39544459+ulixius9@users.noreply.github.com>
Date: Wed, 29 Jan 2025 10:59:16 +0530
Subject: [PATCH] MINOR: Mariadb Lineage Fix (#19575)

---
 .../metadata/ingestion/source/database/mariadb/lineage.py   | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/ingestion/src/metadata/ingestion/source/database/mariadb/lineage.py b/ingestion/src/metadata/ingestion/source/database/mariadb/lineage.py
index afbfcc398f3..074ce5b8c6a 100644
--- a/ingestion/src/metadata/ingestion/source/database/mariadb/lineage.py
+++ b/ingestion/src/metadata/ingestion/source/database/mariadb/lineage.py
@@ -14,7 +14,7 @@ Mariadb lineage module
 from typing import Optional
 
 from metadata.generated.schema.entity.services.connections.database.mariaDBConnection import (
-    MariadbConnection,
+    MariaDBConnection,
 )
 from metadata.generated.schema.metadataIngestion.workflow import (
     Source as WorkflowSource,
@@ -38,8 +38,8 @@ class MariadbLineageSource(LineageSource):
     ):
         """Create class instance"""
         config: WorkflowSource = WorkflowSource.model_validate(config_dict)
-        connection: MariadbConnection = config.serviceConnection.root.config
-        if not isinstance(connection, MariadbConnection):
+        connection: MariaDBConnection = config.serviceConnection.root.config
+        if not isinstance(connection, MariaDBConnection):
             raise InvalidSourceException(
                 f"Expected MariadbConnection, but got {connection}"
             )