diff --git a/tools/demo/mammoth_demo.py b/tools/demo/mammoth_demo.py
index 549c76bb..50cc4fdf 100644
--- a/tools/demo/mammoth_demo.py
+++ b/tools/demo/mammoth_demo.py
@@ -72,8 +72,6 @@
 </style>
 """
 
-
-
 ARCHITECTURE_HTML_ABNEG = """
 <h3>Decoder</h3>
 <div class="arch">
@@ -131,6 +129,7 @@
 </style>
 """
 
+
 def render(template, model_task):
     task, lang = model_task.split('_')
     if task == 'translate' or task == 'train':
@@ -176,7 +175,7 @@ def __call__(self):
                 height=None,
             )
         with col2:
-            architecture_html = ARCHITECTURE_HTML_HYDRA if not 'train' in model.task else ARCHITECTURE_HTML_ABNEG
+            architecture_html = ARCHITECTURE_HTML_HYDRA if 'train' not in model.task else ARCHITECTURE_HTML_ABNEG
             st.markdown(
                 render(architecture_html, model.task),
                 unsafe_allow_html=True,