diff --git a/benchmark/server/main.go b/benchmark/server/main.go
index 090f002f..aa73ffe9 100644
--- a/benchmark/server/main.go
+++ b/benchmark/server/main.go
@@ -28,7 +28,10 @@ func main() {
 			grpclog.Fatalf("Failed to serve: %v", err)
 		}
 	}()
-	addr, stopper := benchmark.StartServer(":0") // listen on all interfaces
+	addr, stopper, err := benchmark.StartServer(benchmark.ServerInfo{Addr: ":0", Type: "protobuf"}) // listen on all interfaces
+	if err != nil {
+		grpclog.Fatalf("failed to start server: %v", err)
+	}
 	grpclog.Println("Server Address: ", addr)
 	<-time.After(time.Duration(*duration) * time.Second)
 	stopper()