diff --git a/x-pack/plugins/lens/public/state_management/context_middleware/index.ts b/x-pack/plugins/lens/public/state_management/context_middleware/index.ts
index a32ae86fa2e3b..6abaf3eff851e 100644
--- a/x-pack/plugins/lens/public/state_management/context_middleware/index.ts
+++ b/x-pack/plugins/lens/public/state_management/context_middleware/index.ts
@@ -26,7 +26,7 @@ function isTimeBased(state: LensState, datasourceMap: DatasourceMap) {
   const { activeDatasourceId, datasourceStates } = state.lens;
   return Boolean(
     activeDatasourceId &&
-      datasourceMap[activeDatasourceId].isTimeBased(datasourceStates[activeDatasourceId].state)
+      datasourceMap[activeDatasourceId].isTimeBased?.(datasourceStates[activeDatasourceId].state)
   );
 }