fix class ref

alexanderthclark · Jun 29, 2024 · eaa5cf1 · eaa5cf1
1 parent a2fa629
commit eaa5cf1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/freeride/curves.py b/freeride/curves.py
@@ -271,7 +271,7 @@ def vertical_shift(self, delta, inplace=True):
         if inplace:
             self.__init__(*coef, symbol=self.symbol)
         else:
-            return cls(*coef, symbol=self.symbol)
+            return self.__class__(*coef, symbol=self.symbol)
 
     def horizontal_shift(self, delta, inplace=True):
         """
@@ -290,7 +290,7 @@ def horizontal_shift(self, delta, inplace=True):
         if inplace:
             self.__init__(*coef, symbol=self.symbol)
         else:
-            return cls(*coef, symbol=self.symbol)
+            return self.__class__(*coef, symbol=self.symbol)
 
     def plot(self, ax=None, textbook_style=True, max_q=100,
              label=True, **kwargs):