diff --git a/client/main.go b/client/main.go
index 4f981bd..39a4bf8 100644
--- a/client/main.go
+++ b/client/main.go
@@ -270,7 +270,7 @@
 	}
 	switch *driver {
 	case "openolt":
-		driverType = api.AddOLTChassisMessage_openoltDriver
+		driverType = api.AddOLTChassisMessage_openolt
 
 	}
 
@@ -310,7 +310,7 @@
 	fmt.Println("tech_profile", *techProfile)
 	fmt.Println("speed_profile", *speedProfile)
 	res, err := c.PreProvisionOnt(context.Background(), &api.PreProvisionOntMessage{CLLI: *clli, SlotNumber: int32(*slot), PortNumber: int32(*port),
-		OntNumber: int32(*ont), STag: uint32(*stag), CTag: uint32(*ctag), TechProfile: *techProfile, SpeedProfile: *speedProfile})
+		OntNumber: int32(*ont), STag: uint32(*stag), CTag: uint32(*ctag), NasPortID: *nasPort, CircuitID: *circuitID, TechProfile: *techProfile, SpeedProfile: *speedProfile})
 	if err != nil {
 		debug.PrintStack()
 		fmt.Printf("Error when calling ProvsionOnt %s", err)