diff --git a/main_test.go b/main_test.go
index f25d132..cffd0aa 100644
--- a/main_test.go
+++ b/main_test.go
@@ -112,8 +112,7 @@
 
 	ad := newMockAdapter()
 	oolt, err := ad.startOpenOLT(context.TODO(), nil,
-		ad.coreProxy, ad.adapterProxy, ad.eventProxy, 1, ad.config.KVStoreHost,
-		ad.config.KVStorePort, ad.config.KVStoreType)
+		ad.coreProxy, ad.adapterProxy, ad.eventProxy, ad.config)
 	if oolt != nil {
 		t.Log("Open OLT ", oolt)
 	}
@@ -167,8 +166,7 @@
 	ad.kip.Start()
 
 	oolt, _ := ad.startOpenOLT(context.TODO(), nil,
-		ad.coreProxy, ad.adapterProxy, ad.eventProxy, 1, ad.config.KVStoreHost,
-		ad.config.KVStorePort, ad.config.KVStoreType)
+		ad.coreProxy, ad.adapterProxy, ad.eventProxy, ad.config)
 	printBanner()
 	printVersion()
 	ctx := context.TODO()