1 files changed, 7 insertions, 7 deletions
diff --git a/tests/test_tokens.py b/tests/test_tokens.py
index d4772ba..1d1b966 100644
--- a/tests/test_tokens.py
+++ b/tests/test_tokens.py
@@ -7,13 +7,13 @@ class TestTokens(unittest.TestCase):
     def test_comment_attachment(self):
         tokenizer = Tokenizer()
         sql_comment = [
-            ("/*comment*/ foo", "comment"),
-            ("/*comment*/ foo --test", "comment"),
-            ("--comment\nfoo --test", "comment"),
-            ("foo --comment", "comment"),
-            ("foo", None),
-            ("foo /*comment 1*/ /*comment 2*/", "comment 1"),
+            ("/*comment*/ foo", ["comment"]),
+            ("/*comment*/ foo --test", ["comment", "test"]),
+            ("--comment\nfoo --test", ["comment", "test"]),
+            ("foo --comment", ["comment"]),
+            ("foo", []),
+            ("foo /*comment 1*/ /*comment 2*/", ["comment 1", "comment 2"]),
         ]
 
         for sql, comment in sql_comment:
-            self.assertEqual(tokenizer.tokenize(sql)[0].comment, comment)
+            self.assertEqual(tokenizer.tokenize(sql)[0].comments, comment)