diff --git a/sklearn_extra/robust/_robust_weighted_estimator_helper.pyx b/sklearn_extra/robust/_robust_weighted_estimator_helper.pyx
index 07a6fca..94d0e4a 100644
--- a/sklearn_extra/robust/_robust_weighted_estimator_helper.pyx
+++ b/sklearn_extra/robust/_robust_weighted_estimator_helper.pyx
@@ -70,7 +70,7 @@ cpdef np.ndarray[floating] _kmeans_loss(np.ndarray[floating, ndim=2, mode='c'] X
         np.ndarray[floating, ndim=2] centers = np.zeros([n_classes,
                                                          n_features],
                                                          dtype = dtype)
-        np.ndarray[int32_t] num_in_cluster = np.zeros(n_classes, dtype = int)
+        np.ndarray[int] num_in_cluster = np.zeros(n_classes, dtype = int)
         np.ndarray[floating] inertias = np.zeros(n_samples, dtype = dtype)
     for i in range(n_samples):
         for j in range(n_features):