diff --git a/x-pack/legacy/plugins/ml/public/contexts/ui/__mocks__/mocks.ts b/x-pack/legacy/plugins/ml/public/contexts/ui/__mocks__/mocks.ts
index 96a110c65935a0..785daec0ab369d 100644
--- a/x-pack/legacy/plugins/ml/public/contexts/ui/__mocks__/mocks.ts
+++ b/x-pack/legacy/plugins/ml/public/contexts/ui/__mocks__/mocks.ts
@@ -38,10 +38,10 @@ const time = {
 
 export const uiTimefilterMock = {
   isAutoRefreshSelectorEnabled() {
-    return this.isAutoRefreshSelectorEnabled;
+    return this._isAutoRefreshSelectorEnabled;
   },
   isTimeRangeSelectorEnabled() {
-    return this.isTimeRangeSelectorEnabled;
+    return this._isTimeRangeSelectorEnabled;
   },
   enableAutoRefreshSelector() {
     this._isAutoRefreshSelectorEnabled = true;