diff --git a/pts/model/n_beats/n_beats_ensemble.py b/pts/model/n_beats/n_beats_ensemble.py
index f32cbbf..7792e90 100644
--- a/pts/model/n_beats/n_beats_ensemble.py
+++ b/pts/model/n_beats/n_beats_ensemble.py
@@ -170,7 +170,7 @@ class NBEATSEnsembleEstimator(PyTorchEstimator):
         Arguments passed down to the individual estimators.
     """
 
-    @validted()
+    @validated()
     def __init__(
         self,
         freq: str,
diff --git a/pts/model/n_beats/n_beats_estimator.py b/pts/model/n_beats/n_beats_estimator.py
index 40f3e5b..a00739e 100644
--- a/pts/model/n_beats/n_beats_estimator.py
+++ b/pts/model/n_beats/n_beats_estimator.py
@@ -30,7 +30,7 @@ from .n_beats_network import (
 )
 
 
-class NBEATSEstimator:
+class NBEATSEstimator(PyTorchEstimator):
     @validated()
     def __init__(
         self,