diff --git a/gordo/machine/model/models.py b/gordo/machine/model/models.py
index f3d7619b6..caa985b8e 100644
--- a/gordo/machine/model/models.py
+++ b/gordo/machine/model/models.py
@@ -19,7 +19,7 @@
 import pandas as pd
 import xarray as xr
 
-from sklearn.base import TransformerMixin, BaseEstimator
+from sklearn.base import TransformerMixin
 from sklearn.metrics import explained_variance_score
 from sklearn.exceptions import NotFittedError
 
@@ -34,7 +34,7 @@
 logger = logging.getLogger(__name__)
 
 
-class KerasBaseEstimator(BaseWrapper, GordoBase, BaseEstimator):
+class KerasBaseEstimator(BaseWrapper, GordoBase):
     supported_fit_args = [
         "batch_size",
         "epochs",