From d033ba7948606a84a074cb7eab792371573ec49e Mon Sep 17 00:00:00 2001
From: Dominic Kramer <DominicDKramer@google.com>
Date: Tue, 14 Feb 2017 15:24:14 -0800
Subject: [PATCH] Update to reflect changes to the plugin API

---
 src/plugins/plugin-redis.js | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/plugins/plugin-redis.js b/src/plugins/plugin-redis.js
index 0bc916b14..2b7cc1e8c 100644
--- a/src/plugins/plugin-redis.js
+++ b/src/plugins/plugin-redis.js
@@ -79,7 +79,7 @@ function startSpanFromArguments(api, cmd, args, cb, send_command) {
 
 function internalSendCommandWrap(api, internal_send_command) {
   return function internal_send_command_trace(cmd, args, cb) {
-    var root = api.getTransaction();
+    var root = api.getRootSpan();
     if (!root) {
       return internal_send_command.call(this, cmd, args, cb);
     }
@@ -94,8 +94,8 @@ function internalSendCommandWrap(api, internal_send_command) {
 
 function sendCommandWrap(api, send_command) {
   return function send_command_trace(cmd, args, cb) {
-    var transaction = api.getTransaction();
-    if (!transaction) {
+    var root = api.getRootSpan();
+    if (!root) {
       return send_command.call(this, cmd, args, cb);
     }
     return startSpanFromArguments(api, cmd, args, cb, send_command.bind(this));