diff --git a/src/galax/potential/_potential/base.py b/src/galax/potential/_potential/base.py
index 1d499a71..69e91938 100644
--- a/src/galax/potential/_potential/base.py
+++ b/src/galax/potential/_potential/base.py
@@ -1,7 +1,7 @@
 __all__ = ["AbstractPotentialBase"]
 
 import abc
-from dataclasses import fields, replace
+from dataclasses import KW_ONLY, fields, replace
 from typing import TYPE_CHECKING, Any
 
 import astropy.units as u
@@ -40,6 +40,7 @@
 class AbstractPotentialBase(eqx.Module, metaclass=ModuleMeta, strict=True):  # type: ignore[misc]
     """Potential Class."""
 
+    _: KW_ONLY
     units: eqx.AbstractVar[UnitSystem]
 
     ###########################################################################