diff --git a/aeon/forecasting/tests/test_all_forecasters.py b/aeon/forecasting/tests/test_all_forecasters.py
index bf5243eecd..8af4a3134d 100644
--- a/aeon/forecasting/tests/test_all_forecasters.py
+++ b/aeon/forecasting/tests/test_all_forecasters.py
@@ -310,8 +310,8 @@ def test_predict_interval(
     ):
         """Check prediction intervals returned by predict.
 
-        Arguments
-        ---------
+        Parameters
+        ----------
         Forecaster: BaseEstimator class descendant, forecaster to test
         fh: ForecastingHorizon, fh at which to test prediction
         coverage: float, coverage at which to make prediction intervals
@@ -375,8 +375,8 @@ def _check_predict_quantiles(
     def test_predict_quantiles(self, estimator_instance, n_columns, fh_int_oos, alpha):
         """Check prediction quantiles returned by predict.
 
-        Arguments
-        ---------
+        Parameters
+        ----------
         Forecaster: BaseEstimator class descendant, forecaster to test
         fh: ForecastingHorizon, fh at which to test prediction
         alpha: float, alpha at which to make prediction intervals
diff --git a/aeon/networks/_ae_fcn.py b/aeon/networks/_ae_fcn.py
index 1cef7d5c31..73d02c6b8f 100644
--- a/aeon/networks/_ae_fcn.py
+++ b/aeon/networks/_ae_fcn.py
@@ -91,8 +91,8 @@ def __init__(
     def build_network(self, input_shape, **kwargs):
         """Construct a network and return its input and output layers.
 
-        Arguments
-        ---------
+        Parameters
+        ----------
         input_shape : tuple of shape = (n_timepoints (m), n_channels (d))
             The shape of the data fed into the input layer.