diff --git a/scrapegraphai/docloaders/chromium.py b/scrapegraphai/docloaders/chromium.py
index 0377f803..7d499245 100644
--- a/scrapegraphai/docloaders/chromium.py
+++ b/scrapegraphai/docloaders/chromium.py
@@ -2,6 +2,7 @@ import asyncio
 import logging
 from typing import Any, AsyncIterator, Iterator, List, Optional
 
+from langchain_community.document_loaders.base import BaseLoader
 from langchain_core.documents import Document
 
 from ..utils import Proxy, dynamic_import, parse_or_search_proxy
@@ -10,7 +11,7 @@ from ..utils import Proxy, dynamic_import, parse_or_search_proxy
 logger = logging.getLogger(__name__)
 
 
-class ChromiumLoader:
+class ChromiumLoader(BaseLoader):
     """scrapes HTML pages from URLs using a (headless) instance of the
     Chromium web driver with proxy protection