diff --git a/src/nifreeze/model/gpr.py b/src/nifreeze/model/gpr.py
index 026867d..45e8b88 100644
--- a/src/nifreeze/model/gpr.py
+++ b/src/nifreeze/model/gpr.py
@@ -25,7 +25,7 @@
 from __future__ import annotations
 
 from numbers import Integral, Real
-from typing import Callable, ClassVar, Mapping, Sequence
+from typing import Callable, ClassVar, Mapping, Optional, Sequence, Union
 
 import numpy as np
 from scipy import optimize
@@ -153,6 +153,8 @@ class DiffusionGPR(GaussianProcessRegressor):
 
     """
 
+    optimizer: Optional[Union[StrOptions, Callable, None]] = None
+
     _parameter_constraints: ClassVar[dict] = {
         "kernel": [None, Kernel],
         "alpha": [Interval(Real, 0, None, closed="left"), np.ndarray],