diff --git a/services/compute/virtualmachine/client.go b/services/compute/virtualmachine/client.go
index 8a162f11..92eac9ab 100644
--- a/services/compute/virtualmachine/client.go
+++ b/services/compute/virtualmachine/client.go
@@ -18,7 +18,7 @@ import (
 type Service interface {
 	Get(context.Context, string, string) (*[]compute.VirtualMachine, error)
 	CreateOrUpdate(context.Context, string, string, *compute.VirtualMachine) (*compute.VirtualMachine, error)
-	Hydrate(context.Context, string, string) (*compute.VirtualMachine, error)
+	Hydrate(context.Context, string, string, *compute.VirtualMachine) (*compute.VirtualMachine, error)
 	Delete(context.Context, string, string) error
 	Query(context.Context, string, string) (*[]compute.VirtualMachine, error)
 	Start(context.Context, string, string) error
@@ -61,8 +61,8 @@ func (c *VirtualMachineClient) CreateOrUpdate(ctx context.Context, group, name s
 }
 
 // Hydrate methods creates MOC representation of the VM resource
-func (c *VirtualMachineClient) Hydrate(ctx context.Context, group, name string) (*compute.VirtualMachine, error) {
-	return c.internal.Hydrate(ctx, group, name)
+func (c *VirtualMachineClient) Hydrate(ctx context.Context, group, name string, compute *compute.VirtualMachine) (*compute.VirtualMachine, error) {
+	return c.internal.Hydrate(ctx, group, name, compute)
 }
 
 // Delete methods invokes delete of the compute resource
diff --git a/services/compute/virtualmachine/wssd.go b/services/compute/virtualmachine/wssd.go
index 4a2f783a..0b993a04 100644
--- a/services/compute/virtualmachine/wssd.go
+++ b/services/compute/virtualmachine/wssd.go
@@ -78,8 +78,8 @@ func (c *client) CreateOrUpdate(ctx context.Context, group, name string, sg *com
 }
 
 // Hydrate
-func (c *client) Hydrate(ctx context.Context, group, name string) (*compute.VirtualMachine, error) {
-	request, err := c.getVirtualMachineRequest(wssdcloudproto.Operation_HYDRATE, group, name, nil)
+func (c *client) Hydrate(ctx context.Context, group, name string, sg *compute.VirtualMachine) (*compute.VirtualMachine, error) {
+	request, err := c.getVirtualMachineRequest(wssdcloudproto.Operation_HYDRATE, group, name, sg)
 	if err != nil {
 		return nil, err
 	}