diff --git a/api/handler.go b/api/handler.go
index 3b1754f..4e1ef61 100644
--- a/api/handler.go
+++ b/api/handler.go
@@ -110,9 +110,10 @@
 func (s *Server) CreateOLTChassis(ctx context.Context, in *AddOLTChassisMessage) (*AddOLTChassisReturn, error) {
 	clli := in.GetCLLI()
 	oltType := in.GetType().String()
+	driver := in.GetDriver().String()
 	address := net.TCPAddr{IP: net.ParseIP(in.GetSlotIP()), Port: int(in.GetSlotPort())}
 	hostname := in.GetHostname()
-	clli, err := impl.CreateOLTChassis(clli, oltType, address, hostname)
+	clli, err := impl.CreateOLTChassis(clli, oltType, driver, address, hostname)
 	return &AddOLTChassisReturn{DeviceID: hostname, ChassisDeviceID: clli}, err
 }