diff --git a/lb.go b/lb.go
index 6b840ece08b6fad8944d6d38e280fe543eb661fa..8b441124403ded2cbcb03f793de7dc8cb865b3c8 100644
--- a/lb.go
+++ b/lb.go
@@ -190,6 +190,8 @@ type Loadbalancer struct {
 
 	mx    sync.Mutex
 	nodes map[string]*node
+
+	lbpb.UnimplementedLoadBalancerServer
 }
 
 // New returns a new Loadbalancer with the given utilization limits