diff --git a/src/test/test_trainer.py b/src/test/test_trainer.py
index 3b8ad72..0d9cd14 100644
--- a/src/test/test_trainer.py
+++ b/src/test/test_trainer.py
@@ -29,5 +29,4 @@ def test_predict(self):
         self.assertEqual(TokenType.TITLE, tokens[0].token_type)
         self.assertEqual("Document Big Centered Title", tokens[0].content)
         self.assertEqual(TokenType.TEXT, tokens[1].token_type)
-        self.assertEqual(TokenType.TEXT, tokens[10].token_type)
         self.assertEqual("List Title", tokens[10].content)