1 files changed, 2 insertions, 2 deletions
diff --git a/spice.go b/spice.go
index 8443539..b426d6d 100644
--- a/spice.go
+++ b/spice.go
@@ -6,11 +6,11 @@ import (
 	"errors"
 	"fmt"
 
-	pb "go.wit.com/lib/protobuf/virtbuf"
+	"go.wit.com/lib/protobuf/virtpb"
 	"libvirt.org/go/libvirtxml"
 )
 
-func SetSpicePort(d *pb.Droplet, domcfg *libvirtxml.Domain) error {
+func SetSpicePort(d *virtpb.Droplet, domcfg *libvirtxml.Domain) error {
 	if domcfg.Devices.Graphics == nil {
 		return errors.New("no graphics")
 	}