From 4c1c1f39fe6a730a8f8a948e66e9c2a96a71d873 Mon Sep 17 00:00:00 2001
From: "Joshua Z. Zhang" <cheungchih@gmail.com>
Date: Fri, 11 May 2018 15:47:41 -0700
Subject: [PATCH] fix test

---
 src/engine/threaded_engine_pooled.cc     | 6 +++---
 tests/cpp/engine/threaded_engine_test.cc | 4 ++--
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/engine/threaded_engine_pooled.cc b/src/engine/threaded_engine_pooled.cc
index 6025a174a93e..374357126e47 100644
--- a/src/engine/threaded_engine_pooled.cc
+++ b/src/engine/threaded_engine_pooled.cc
@@ -57,9 +57,9 @@ class ThreadedEnginePooled : public ThreadedEngine {
     task_queue_->SignalForKill();
     task_queue_ = nullptr;
     io_task_queue_->SignalForKill();
-    io_task_queue = nullptr;
-    thread_pool_.reset(nullptr);
-    io_thread_pool = nullptr;
+    io_task_queue_ = nullptr;
+    thread_pool_ = nullptr;
+    io_thread_pool_ = nullptr;
   }
 
   void Stop() override {
diff --git a/tests/cpp/engine/threaded_engine_test.cc b/tests/cpp/engine/threaded_engine_test.cc
index 52dd40e05c66..6428ed80dd69 100644
--- a/tests/cpp/engine/threaded_engine_test.cc
+++ b/tests/cpp/engine/threaded_engine_test.cc
@@ -129,8 +129,8 @@ TEST(Engine, start_stop) {
   engine[2] = mxnet::engine::CreateThreadedEnginePerDevice();
 
   for (int i = 0; i < num_engine; ++i) {
-    engine[i].stop();
-    engine[i].start();
+    engine[i]->Stop();
+    engine[i]->Start();
   }
 }