diff --git a/tests/e2e/test_client_e2e.py b/tests/e2e/test_client_e2e.py
index e250845..c6288a3 100644
--- a/tests/e2e/test_client_e2e.py
+++ b/tests/e2e/test_client_e2e.py
@@ -1,3 +1,4 @@
+import os
 import time
 from unittest.mock import Mock, patch
 
@@ -24,6 +25,10 @@ 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.
@@ -31,12 +36,10 @@ def test_list_nodes_category(self, e2e_client: Client):
         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):
         """