Fix failing add node tests after category change

tests/e2e/test_client_e2e.py

@@ -1,3 +1,4 @@
+import os
 import time
 from unittest.mock import Mock, patch
 
@@ -24,19 +25,21 @@ def test_list_nodes(self, e2e_client: Client):
         """
         e2e_client.list_nodes()
 
+    @pytest.mark.skipif(
+        os.getenv("UE_ENVIRONMENT") == "prod",
+        reason="Mathematical Operators category is 'Basic' in prod until we deploy there",
+    )
     def test_list_nodes_category(self, e2e_client: Client):
         """
         Verify that nodes can be filtered by category.
 
         Args:
             e2e_client: A Client instance.
         """
-        respone = e2e_client.list_nodes(category="Basic")
+        response = e2e_client.list_nodes(category="Mathematical Operators")
 
-        # There should only be one node in the demo category
-        assert len(respone) == 1
-        assert respone[0]["category"] == "Basic"
-        assert respone[0]["id"] == "Add"
+        assert len(response) > 0
+        assert all(node["category"] == "Mathematical Operators" for node in response)
 
     def test_queue_node(self, e2e_client: Client):
         """