diff --git a/python/examples/grpc_impl_example/fit_a_line/test_timeout_client.py b/python/examples/grpc_impl_example/fit_a_line/test_timeout_client.py
index 5824c301135764e602c8158f12404c54b94f7013..f90fab38533aabf3daa7627ee0b79c56892444dd 100644
--- a/python/examples/grpc_impl_example/fit_a_line/test_timeout_client.py
+++ b/python/examples/grpc_impl_example/fit_a_line/test_timeout_client.py
@@ -28,8 +28,7 @@ for i in range(3):
     fetch_map = client.predict(feed={"x": x}, fetch=["price"])
     if fetch_map["serving_status_code"] == 0:
         print(fetch_map)
-    elif fetch_map["serving_status_code"].code(
-    ) == grpc.StatusCode.DEADLINE_EXCEEDED:
+    elif fetch_map["serving_status_code"] == grpc.StatusCode.DEADLINE_EXCEEDED:
         print('timeout')
     else:
         print(fetch_map["serving_status_code"])