diff --git a/tests/api/test_udf.py b/tests/api/test_udf.py
index 1566e1e6..515ff283 100644
--- a/tests/api/test_udf.py
+++ b/tests/api/test_udf.py
@@ -1,35 +1,50 @@
-
 from pprint import pprint
-
 import pytest
 
 
-@pytest.fixture(scope="module", autouse=True)
-def discovery(npu):
-    npu.objects_discovery()
-
-
-@pytest.fixture(scope="module", autouse=True)
+@pytest.fixture(scope='module', autouse=True)
 def skip_all(testbed_instance):
     testbed = testbed_instance
     if testbed is not None and len(testbed.npu) != 1:
-        pytest.skip("invalid for {} testbed".format(testbed.name))
-
+        pytest.skip('invalid for {} testbed'.format(testbed.name))
 
 @pytest.mark.npu
 class TestSaiUdf:
     # object with parent SAI_OBJECT_TYPE_UDF_MATCH SAI_OBJECT_TYPE_UDF_GROUP
 
     def test_udf_create(self, npu):
-
-        commands = [{'name': 'udf_match_1', 'op': 'create', 'type': 'SAI_OBJECT_TYPE_UDF_MATCH', 'attributes': []}, {'name': 'udf_group_1', 'op': 'create', 'type': 'SAI_OBJECT_TYPE_UDF_GROUP', 'attributes': ['SAI_UDF_GROUP_ATTR_LENGTH', '10']}, {'name': 'udf_1', 'op': 'create', 'type': 'SAI_OBJECT_TYPE_UDF', 'attributes': ['SAI_UDF_ATTR_MATCH_ID', '$udf_match_1', 'SAI_UDF_ATTR_GROUP_ID', '$udf_group_1', 'SAI_UDF_ATTR_OFFSET', '10']}]
+        commands = [
+            {
+                'name': 'udf_match_1',
+                'op': 'create',
+                'type': 'SAI_OBJECT_TYPE_UDF_MATCH',
+                'attributes': [],
+            },
+            {
+                'name': 'udf_group_1',
+                'op': 'create',
+                'type': 'SAI_OBJECT_TYPE_UDF_GROUP',
+                'attributes': ['SAI_UDF_GROUP_ATTR_LENGTH', '10'],
+            },
+            {
+                'name': 'udf_1',
+                'op': 'create',
+                'type': 'SAI_OBJECT_TYPE_UDF',
+                'attributes': [
+                    'SAI_UDF_ATTR_MATCH_ID',
+                    '$udf_match_1',
+                    'SAI_UDF_ATTR_GROUP_ID',
+                    '$udf_group_1',
+                    'SAI_UDF_ATTR_OFFSET',
+                    '10',
+                ],
+            },
+        ]
 
         results = [*npu.process_commands(commands)]
         print('======= SAI commands RETURN values create =======')
         pprint(results)
 
-
-
     @pytest.mark.dependency(name="test_sai_udf_attr_base_set")
     def test_sai_udf_attr_base_set(self, npu):
 
@@ -66,47 +81,13 @@ def test_sai_udf_attr_base_get(self, npu):
         assert r_value == 'SAI_UDF_BASE_L2', 'Get error, expected SAI_UDF_BASE_L2 but got %s' %  r_value
 
 
-    @pytest.mark.dependency(name="test_sai_udf_attr_hash_mask_set")
-    def test_sai_udf_attr_hash_mask_set(self, npu):
-
-        commands = [
-            {
-                "name": "udf_1",
-                "op": "set",
-                "attributes": ["SAI_UDF_ATTR_HASH_MASK", 'const']
-            }
-        ]
-        results = [*npu.process_commands(commands)]
-        print("======= SAI commands RETURN values set =======")
-        pprint(results)
-
-
-
-    @pytest.mark.dependency(depends=["test_sai_udf_attr_hash_mask_set"])
-    def test_sai_udf_attr_hash_mask_get(self, npu):
-
+    def test_udf_remove(self, npu):
         commands = [
-            {
-                "name": "udf_1",
-                "op": "get",
-                "attributes": ["SAI_UDF_ATTR_HASH_MASK"]
-            }
+            {'name': 'udf_1', 'op': 'remove'},
+            {'name': 'udf_group_1', 'op': 'remove'},
+            {'name': 'udf_match_1', 'op': 'remove'},
         ]
-        results = [*npu.process_commands(commands)]
-        print("======= SAI commands RETURN values get =======")
-        for command in results:
-            for attribute in command:
-                pprint(attribute.raw())
-        r_value = results[0][0].value()
-        print(r_value)
-        assert r_value == 'const', 'Get error, expected const but got %s' %  r_value
-
-
-    def test_udf_remove(self, npu):
-
-        commands = [{'name': 'udf_1', 'op': 'remove'}, {'name': 'udf_group_1', 'op': 'remove'}, {'name': 'udf_match_1', 'op': 'remove'}]
 
         results = [*npu.process_commands(commands)]
         print('======= SAI commands RETURN values remove =======')
-        pprint(results)
-
+        pprint(results)
\ No newline at end of file