From 04b79ade4736368c46e5d6bc7391c58af4a01f9b Mon Sep 17 00:00:00 2001
From: Nathan Kellenicki <nathan@kellenicki.com>
Date: Mon, 16 Dec 2019 21:21:25 -0800
Subject: [PATCH] Renamed devices

---
 src/devices/colordistancesensor.ts | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/devices/colordistancesensor.ts b/src/devices/colordistancesensor.ts
index 84018e6..936109b 100644
--- a/src/devices/colordistancesensor.ts
+++ b/src/devices/colordistancesensor.ts
@@ -36,12 +36,17 @@ export class ColorDistanceSensor extends Device {
 
         switch (mode) {
             case ColorDistanceSensor.Mode.COLOR:
-                if (this._isWeDo2 && message[2] <= 10) {
-                    const color = message[2];
+                if (message[this._isWeDo2 ? 2 : 4] <= 10) {
+                    const color = message[this._isWeDo2 ? 2 : 4];
                     this.emit("color", color);
                 }
                 break;
+
             case ColorDistanceSensor.Mode.COLOR_AND_DISTANCE:
+                if (this._isWeDo2) {
+                    break;
+                }
+
                 /**
                  * Emits when a color sensor is activated.
                  * @event ColorDistanceSensor#color
@@ -82,6 +87,7 @@ export class ColorDistanceSensor extends Device {
                     this.emit("colorAndDistance", color, distance);
                 }
                 break;
+
         }
     }