diff --git a/.ipynb_checkpoints/main-checkpoint.py b/.ipynb_checkpoints/main-checkpoint.py
index 269fda9..5249003 100644
--- a/.ipynb_checkpoints/main-checkpoint.py
+++ b/.ipynb_checkpoints/main-checkpoint.py
@@ -25,9 +25,9 @@
 
 def handle(conf):
     # paste your code here
-    if conf['style'] == ['藏头诗']:
+    if conf['style'] == '藏头诗':
         poetry = writer.cangtou(conf['Chinese_word'])
-    elif conf['style'] == ['藏字诗']:
+    elif conf['style'] == '藏字诗':
         poetry = writer.hide_words(conf['Chinese_word'])
     else:
         poetry = writer.rhyme_verse()
diff --git a/main.py b/main.py
index 269fda9..5249003 100644
--- a/main.py
+++ b/main.py
@@ -25,9 +25,9 @@
 
 def handle(conf):
     # paste your code here
-    if conf['style'] == ['藏头诗']:
+    if conf['style'] == '藏头诗':
         poetry = writer.cangtou(conf['Chinese_word'])
-    elif conf['style'] == ['藏字诗']:
+    elif conf['style'] == '藏字诗':
         poetry = writer.hide_words(conf['Chinese_word'])
     else:
         poetry = writer.rhyme_verse()