1 files changed, 3 insertions, 3 deletions
diff --git a/app/relay/runner.js b/app/relay/runner.js
index 85f08ca..a449a6c 100644
--- a/app/relay/runner.js
+++ b/app/relay/runner.js
@@ -347,11 +347,11 @@ export function run_next_task(){
 }
 
 export function stop_task(task, sigkill){
-  if (!task) return
-  if (task === 'cpu' || state.current_cpu_task.task.uuid === task.uuid) {
+  if (!task) return { error: 'no such task' }
+  if (task.processor === 'cpu' || state.current_cpu_task.task.uuid === task.uuid) {
     terminate(state.current_cpu_task, sigkill)
     return { status: 'ok' }
-  } else if (task === 'gpu' || state.current_gpu_task.task.uuid === task.uuid) {
+  } else if (task.processor === 'gpu' || state.current_gpu_task.task.uuid === task.uuid) {
     terminate(state.current_gpu_task, sigkill)
     return { status: 'ok' }
   }