diff --git a/pace.py b/pace.py
index f555296..fb3c476 100644
--- a/pace.py
+++ b/pace.py
@@ -5,7 +5,7 @@
 import matplotlib.pyplot as plt
 import torch
 
-# We depend on Erfi function, but python.special currently has no implementation.
+# We depend on Erfi function, but torch.special currently has no implementation.
 # We instead modify and rely on https://github.com/redsnic/torch_erf
 
 def polyval(x,coeffs):
@@ -294,4 +294,4 @@ def step(self):
 
     PACEOPT.__name__ += Base.__name__
 
-    return PACEOPT
\ No newline at end of file
+    return PACEOPT