diff --git a/source/Mlos.Python/mlos/Spaces/unit_tests/TestSortedBinaryTree.py b/source/Mlos.Python/mlos/Spaces/unit_tests/TestSortedBinaryTree.py
index 6a0c47bb58..d864857e6c 100644
--- a/source/Mlos.Python/mlos/Spaces/unit_tests/TestSortedBinaryTree.py
+++ b/source/Mlos.Python/mlos/Spaces/unit_tests/TestSortedBinaryTree.py
@@ -9,7 +9,7 @@
 NUM_KEYS = 10000
 RANGE = 100 * NUM_KEYS
 
-class StackTest:
+class TestStack:
 
     def test_stack(self):
         stack = Stack()
@@ -26,7 +26,7 @@ def test_stack(self):
             assert True
 
 
-class SortedBinaryTreeTest:
+class TestSortedBinaryTree:
 
     def setup_method(self, method):
         self.random_sorted_binary_tree = SortedBinaryTree()