diff --git a/text/japanese_bert.py b/text/japanese_bert.py
index 4f349df..d5bfa26 100644
--- a/text/japanese_bert.py
+++ b/text/japanese_bert.py
@@ -4,7 +4,7 @@ from transformers import AutoTokenizer, AutoModelForMaskedLM
 
 tokenizer = AutoTokenizer.from_pretrained("./bert/bert-base-japanese-v3")
 
-def get_bert_feature(text, word2ph):
+def get_bert_feature(text, word2ph, device=None):
     if sys.platform == "darwin" and torch.backends.mps.is_available() and device == "cpu":
         device = "mps"
     if not device: