diff --git a/adaptercore/device_handler_test.go b/adaptercore/device_handler_test.go
index dcb1f17..aacc930 100644
--- a/adaptercore/device_handler_test.go
+++ b/adaptercore/device_handler_test.go
@@ -136,8 +136,8 @@
 		Root:     true,
 		ParentId: "logical_device",
 		Ports: []*voltha.Port{
-			{PortNo: 1, Label: "pon"},
-			{PortNo: 2, Label: "nni"},
+			{PortNo: 1, Label: "pon", Type: voltha.Port_PON_OLT},
+			{PortNo: 2, Label: "nni", Type: voltha.Port_ETHERNET_NNI},
 		},
 		ProxyAddress: &voltha.Device_ProxyAddress{
 			DeviceId:       "olt",