diff --git a/pysindy/utils/base.py b/pysindy/utils/base.py
index 415fd483..ea3cba62 100644
--- a/pysindy/utils/base.py
+++ b/pysindy/utils/base.py
@@ -232,7 +232,7 @@ def get_prox(
 
 
 def regularization_l0(x: NDArray[np.float64], lam: NDArray[np.float64]):
-    return lam.item() * np.count_nonzero(x)
+    return lam * np.count_nonzero(x)
 
 
 def regualization_weighted_l0(x: NDArray[np.float64], lam: NDArray[np.float64]):