1 files changed, 2 insertions, 4 deletions
diff --git a/helpers.go b/helpers.go
index 01ac4b6..faef03e 100644
--- a/helpers.go
+++ b/helpers.go
@@ -89,16 +89,14 @@ func (c *Cluster) AddHypervisor(hostname string, cpus int, mem int) *Hypervisor
 	return h
 }
 
-func (c *Cluster) AddDroplet(hostname string, cpus int, mem int) *Droplet {
+func (c *Cluster) AddDroplet(uuid string, hostname string, cpus int, mem int) *Droplet {
 	d := c.FindDroplet(hostname)
 	if d != nil {
 		return d
 	}
 
-	// Generate a new UUID
-	id := uuid.New()
 	d = &Droplet{
-		Uuid:     id.String(),
+		Uuid:     uuid,
 		Hostname: hostname,
 		Cpus:     int64(cpus),
 	}