diff --git a/client/python/armada_client/_proto_methods.py b/client/python/armada_client/_proto_methods.py
index 608527842f1..ab8bd65fbf6 100644
--- a/client/python/armada_client/_proto_methods.py
+++ b/client/python/armada_client/_proto_methods.py
@@ -20,7 +20,7 @@ def is_terminal(self) -> bool:
         JobState.CANCELLED,
         JobState.PREEMPTED,
     }
-    return self.value in terminal_states
+    return self in terminal_states
 
 
 def is_active(self) -> bool:
@@ -35,4 +35,4 @@ def is_active(self) -> bool:
     :returns: True if the job state is active, False if it is terminal.
     :rtype: bool
     """
-    return not is_terminal(self.value)
+    return not is_terminal(self)
diff --git a/client/python/pyproject.toml b/client/python/pyproject.toml
index 587c1f7ab00..ab4cdc84a83 100644
--- a/client/python/pyproject.toml
+++ b/client/python/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "armada_client"
-version = "0.3.3"
+version = "0.3.4"
 description = "Armada gRPC API python client"
 readme = "README.md"
 requires-python = ">=3.7"