switched to pb.DropletState enum

Signed-off-by: Jeff Carr <[email protected]>

1 files changed, 3 insertions, 2 deletions

diff --git a/event.go b/event.go
index a030736..4358ab6 100644
--- a/event.go
+++ b/event.go
@@ -6,6 +6,7 @@ import (
 	"time"
 
 	"go.wit.com/lib/gui/shell"
+	pb "go.wit.com/lib/protobuf/virtbuf"
 	"go.wit.com/log"
 )
 
@@ -43,7 +44,7 @@ func clusterReady() (bool, string) {
 }
 
 func (d *DropletT) dropletReady() (bool, string) {
-	if d.CurrentState == "ON" {
+	if d.CurrentState == pb.DropletState_ON {
 		return false, "EVENT start droplet is already ON"
 	}
 	if d.starts > 2 {
@@ -86,7 +87,7 @@ func Start(name string) (bool, string) {
 		return false, result
 	}
 
-	if d.CurrentState == "ON" {
+	if d.CurrentState == pb.DropletState_ON {
 		return false, "EVENT start droplet is already ON"
 	}