diff --git a/K_means_with_Iris_Data.ipynb b/K_means_with_Iris_Data.ipynb
index 5c8036f..744c755 100644
--- a/K_means_with_Iris_Data.ipynb
+++ b/K_means_with_Iris_Data.ipynb
@@ -231,7 +231,7 @@
    },
    "outputs": [],
    "source": [
-    "x = pd.DataFrame(iris.data, columns=['Sepal Length', 'Sepal Width', 'Petal Length', 'Petal Width'])\n",
+    "x = pd.DataFrame(iris.data, columns=iris.feature_names)\n",
     "y = pd.DataFrame(iris.target, columns=['Target'])"
    ]
   },