diff --git a/tests/test_typical.py b/tests/test_typical.py
index 3d3931e..6ae31ef 100644
--- a/tests/test_typical.py
+++ b/tests/test_typical.py
@@ -13,16 +13,11 @@ def test_2d():
     assert fcm.cluster_centers_[0][1] == pytest.approx(1.66666667, 0.1)
     testing_data = np.array([[0, 1.9], [3, 3], [4, 4], [8, 9], [9.5, 6.5]])
     predicted_membership = fcm.predict(testing_data)
-    # actual_membership = np.array([[0.98777232, 0.01222768],
-    #                               [0.94884591, 0.05115409],
-    #                               [0.82813688, 0.17186312],
-    #                               [0.02482074, 0.97517926],
-    #                               [0.0908581, 0.9091419]])
     actual_membership = np.array([[0.98777232, 0.01222768],
                                   [0.94884591, 0.05115409],
                                   [0.82813688, 0.17186312],
-                                  [0.02482074, 0.90517926],
-                                  [0.0908581, 0.9491419]])
+                                  [0.02482074, 0.97517926],
+                                  [0.0908581, 0.9091419]])
     assert predicted_membership == pytest.approx(actual_membership, 0.01)