diff --git a/src/bayesvalidrox/surrogate_models/orthogonal_matching_pursuit.py b/src/bayesvalidrox/surrogate_models/orthogonal_matching_pursuit.py
index 64b7f102ea7b326eb09009fe6955718416ebed48..d4f99b8a19bb5dbdf41f093bd454c80c63a321bb 100644
--- a/src/bayesvalidrox/surrogate_models/orthogonal_matching_pursuit.py
+++ b/src/bayesvalidrox/surrogate_models/orthogonal_matching_pursuit.py
@@ -195,8 +195,9 @@ class OrthogonalMatchingPursuit(LinearModel, RegressorMixin):
             r = R / norm_R
 
             # Update counters and early-stop criterions
-            countinf = max(0, round(count-k*kmax))
+            countinf = max(0, int(count-k*kmax))
             LOOmin = min(LOOmin, LOO[count])
+
             if count == 0:
                 cond_early = (LOO[0] <= LOOmin)
             else: