diff --git a/stree/Strees.py b/stree/Strees.py
index ecdc1df..0e8b4cb 100644
--- a/stree/Strees.py
+++ b/stree/Strees.py
@@ -192,10 +192,9 @@ class Stree(BaseEstimator, ClassifierMixin):
         if self.use_predictions:
             res = np.expand_dims(node._clf.decision_function(data), 1)
         else:
-            """doesn't work with multiclass as each sample has to do inner
-            product with its own coefficients computes positition of every
-            sample is w.r.t. the hyperplane
-            """
+            # doesn't work with multiclass as each sample has to do inner
+            # product with its own coefficients computes positition of every
+            # sample is w.r.t. the hyperplane
             res = self._linear_function(data, node)
         return res
 
diff --git a/stree/tests/Strees_grapher_test.py b/stree/tests/Strees_grapher_test.py
index cd47593..e702dcc 100644
--- a/stree/tests/Strees_grapher_test.py
+++ b/stree/tests/Strees_grapher_test.py
@@ -95,9 +95,6 @@ class Stree_grapher_test(unittest.TestCase):
             num_figures_after = plt.gcf().number
         self.assertEqual(7, num_figures_after - num_figures_before)
 
-    def test_filt_4_dims_dataset(self):
-        self._clf.fit(*get_dataset(self._random_state, n_features=4))
-
 
 class Snode_graph_test(unittest.TestCase):
     def __init__(self, *args, **kwargs):