Index: branches/pyyaml3000/tests/test_tokens.py
===================================================================
--- branches/pyyaml3000/tests/test_tokens.py	(revision 46)
+++ branches/pyyaml3000/tests/test_tokens.py	(revision 47)
@@ -55,5 +55,5 @@
             scanner = Scanner(Reader(file(data_filename, 'rb')))
             tokens1 = []
-            while not isinstance(scanner.peek_token(), EndToken):
+            while not isinstance(scanner.peek_token(), StreamEndToken):
                 tokens1.append(scanner.get_token())
             tokens1 = [self.replaces[t.__class__] for t in tokens1]
@@ -77,5 +77,5 @@
                 scanner = Scanner(Reader(file(filename, 'rb')))
                 tokens = []
-                while not isinstance(scanner.peek_token(), EndToken):
+                while not isinstance(scanner.peek_token(), StreamEndToken):
                     tokens.append(scanner.get_token().__class__.__name__)
             except: