diff --git a/python/ray/tests/test_component_failures.py b/python/ray/tests/test_component_failures.py
index e55793ab4..f2b55208e 100644
--- a/python/ray/tests/test_component_failures.py
+++ b/python/ray/tests/test_component_failures.py
@@ -2,7 +2,6 @@ from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
-import json
 import os
 import signal
 import sys
@@ -312,12 +311,8 @@ def check_components_alive(cluster, component_type, check_component_alive):
 @pytest.mark.parametrize(
     "ray_start_cluster", [{
         "num_cpus": 8,
-        "num_nodes": 4,
-        "_internal_config": json.dumps({
-            "num_heartbeats_timeout": 100
-        }),
-    }],
-    indirect=True)
+        "num_nodes": 4
+    }], indirect=True)
 def test_raylet_failed(ray_start_cluster):
     cluster = ray_start_cluster
     # Kill all raylets on worker nodes.
@@ -334,12 +329,8 @@ def test_raylet_failed(ray_start_cluster):
 @pytest.mark.parametrize(
     "ray_start_cluster", [{
         "num_cpus": 8,
-        "num_nodes": 2,
-        "_internal_config": json.dumps({
-            "num_heartbeats_timeout": 100
-        }),
-    }],
-    indirect=True)
+        "num_nodes": 2
+    }], indirect=True)
 def test_plasma_store_failed(ray_start_cluster):
     cluster = ray_start_cluster
     # Kill all plasma stores on worker nodes.
diff --git a/src/ray/rpc/raylet/raylet_client.cc b/src/ray/rpc/raylet/raylet_client.cc
index 83272fae1..c6cfc2f05 100644
--- a/src/ray/rpc/raylet/raylet_client.cc
+++ b/src/ray/rpc/raylet/raylet_client.cc
@@ -77,18 +77,10 @@ ray::Status RayletClient::SubmitTask(const ray::TaskSpecification &task_spec) {
   SubmitTaskRequest submit_task_request;
   submit_task_request.mutable_task_spec()->CopyFrom(task_spec.GetMessage());
 
-  auto callback = [this](const Status &status, const SubmitTaskReply &reply) {
-    if (!status.ok() && is_connected_) {
-      is_connected_ = false;
-      RAY_LOG(INFO) << "Failed to send SubmitTaskRequest, msg: " << status.message();
-    }
-  };
-
-  auto call =
-      client_call_manager_.CreateCall<RayletService, SubmitTaskRequest, SubmitTaskReply>(
-          *stub_, &RayletService::Stub::PrepareAsyncSubmitTask, submit_task_request,
-          callback);
-  return call->GetStatus();
+  grpc::ClientContext context;
+  SubmitTaskReply reply;
+  auto status = stub_->SubmitTask(&context, submit_task_request, &reply);
+  return GrpcStatusToRayStatus(status);
 }
 
 ray::Status RayletClient::GetTask(std::unique_ptr<ray::TaskSpecification> *task_spec) {