diff --git a/cmd/openolt-adapter/main.go b/cmd/openolt-adapter/main.go
index f78529b..31c71d5 100644
--- a/cmd/openolt-adapter/main.go
+++ b/cmd/openolt-adapter/main.go
@@ -275,7 +275,7 @@
 	// TODO:  More cleanup
 }
 
-func newKVClient(storeType, address string, timeout int) (kvstore.Client, error) {
+func newKVClient(storeType, address string, timeout time.Duration) (kvstore.Client, error) {
 
 	logger.Infow("kv-store-type", log.Fields{"store": storeType})
 	switch storeType {