diff --git a/ProteinDT/models/model_SDE.py b/ProteinDT/models/model_SDE.py
index 45d0885..b14877e 100644
--- a/ProteinDT/models/model_SDE.py
+++ b/ProteinDT/models/model_SDE.py
@@ -47,7 +47,7 @@ def prior_logp(self, z):
 
     def discretize(self, x, t):
         """Discretize the SDE in the form: x_{i+1} = x_i + f_i(x_i) + G_i z_i.
-        Useful for reverse diffusion sampling and probabiliy flow sampling.
+        Useful for reverse diffusion sampling and probability flow sampling.
         Defaults to Euler-Maruyama discretization.
         Args:
             x: a torch tensor