diff --git a/machine_learning/k_means_clust.py b/machine_learning/k_means_clust.py
index 19e510b966b6..6dd3f9ef916a 100644
--- a/machine_learning/k_means_clust.py
+++ b/machine_learning/k_means_clust.py
@@ -160,7 +160,7 @@ def plot_kmeans(data, centroids, cluster_assignment) -> None:
     ax = plt.axes(projection="3d")
     ax.scatter(data[:, 0], data[:, 1], data[:, 2], c=cluster_assignment, cmap="viridis")
     ax.scatter(
-        centroids[:, 0], centroids[:, 1], centroids[:, 2], c="red", s=100, marker="x"
+        centroids[:, 0], centroids[:, 1], zs=centroids[:, 2], c="red", s=100, marker="x"
     )
     ax.set_xlabel("X")
     ax.set_ylabel("Y")