diff --git a/tsimcne/losses/infonce.py b/tsimcne/losses/infonce.py
index e113a6e..61428db 100644
--- a/tsimcne/losses/infonce.py
+++ b/tsimcne/losses/infonce.py
@@ -136,7 +136,7 @@ def compute(self):
         self.criterion = self.cls(**self.kwargs)
 
 
-class InfoNCET(InfoNCEZ):
+class InfoNCET(InfoNCEGaussian):
     def __init__(self, dof=None, **kwargs):
         super().__init__(**kwargs)
         self.dof = dof