diff --git a/internal/pkg/common/interfaces.go b/internal/pkg/common/interfaces.go
index e761a3d..0465398 100755
--- a/internal/pkg/common/interfaces.go
+++ b/internal/pkg/common/interfaces.go
@@ -192,6 +192,7 @@
 	GetAllBidirectionalGemPortIDsForOnu() []uint16
 	GetNumberOfConfiguredUsGemPorts(ctx context.Context) int
 	SetProfileToDelete(uint8, uint8, bool)
+	GetGEMportToAllocIDMappingForONU(ctx context.Context, aDeviceID string) map[uint16]uint16
 }
 
 // IuniVlanConfigFsm interface to uniVlanConfigFsm