diff --git a/python/whylogs/api/logger/experimental/logger/actor/data_logger.py b/python/whylogs/api/logger/experimental/logger/actor/data_logger.py
index dc8b92381..65ee6db63 100644
--- a/python/whylogs/api/logger/experimental/logger/actor/data_logger.py
+++ b/python/whylogs/api/logger/experimental/logger/actor/data_logger.py
@@ -1,11 +1,7 @@
 from abc import abstractmethod
 from typing import Any, Dict, Generic, List, Optional, TypeVar, Union
 
-# TODO: stubs?
-try:
-    import pandas as pd  # type: ignore
-except ImportError:
-    pd: Any = None  # type: ignore
+from whylogs.core.stubs import pd
 
 
 # TODO add strong typing here. whylogs takes pretty much anything in log() so we don't actually