diff --git a/kmodes/tests/test_kmodes.py b/kmodes/tests/test_kmodes.py
index cd84547..2188bee 100644
--- a/kmodes/tests/test_kmodes.py
+++ b/kmodes/tests/test_kmodes.py
@@ -571,8 +571,7 @@ def test_k_modes_sample_weight_unchanged(self):
     def test_kmodes_fit_predict(self):
         """Test whether fit_predict interface works the same as fit and predict."""
         kmodes = KModes(n_clusters=4, init='Cao', random_state=42)
-        sample_weight = np.zeros(TEST_DATA.shape[0])
-        sample_weight[0] = 1
+        sample_weight = [0.5] * TEST_DATA.shape[0]
         data1 = kmodes.fit_predict(TEST_DATA, sample_weight=sample_weight)
         data2 = kmodes.fit(TEST_DATA, sample_weight=sample_weight).predict(TEST_DATA)
         assert_cluster_splits_equal(data1, data2)