diff --git a/tf_adapter/python/npu_bridge/embedding/embedding_utils.py b/tf_adapter/python/npu_bridge/embedding/embedding_utils.py
index 9d8680f9a02e46d6f4a2664cf4acbc1fa8e74f2f..bd5aef85bfd5c0a47cfb03f0a415662d5d86ab90 100644
--- a/tf_adapter/python/npu_bridge/embedding/embedding_utils.py
+++ b/tf_adapter/python/npu_bridge/embedding/embedding_utils.py
@@ -16,6 +16,7 @@
 # ==============================================================================
 
 import re
+import tensorflow as tf
 
 
 class EmbeddingVariableOption: