diff --git a/internal/pkg/common/interfaces.go b/internal/pkg/common/interfaces.go
index ef5e004..1f7d457 100755
--- a/internal/pkg/common/interfaces.go
+++ b/internal/pkg/common/interfaces.go
@@ -178,5 +178,5 @@
 
 // IuniVlanConfigFsm interface to uniVlanConfigFsm
 type IuniVlanConfigFsm interface {
-	IsFlowRemovePending(chan<- bool) bool
+	IsFlowRemovePending(context.Context, chan<- bool) bool
 }