diff --git a/tests/test_pretty.py b/tests/test_pretty.py
index 3985361..6cc7a89 100644
--- a/tests/test_pretty.py
+++ b/tests/test_pretty.py
@@ -20,14 +20,14 @@ class TestPretty(unittest.TestCase):
         decoder = endecoder.EnDecoder()
         bibdata = decoder.decode_bibdata(bibtex_raw0)
         line = u'Page, Lawrence et al. "The PageRank Citation Ranking: Bringing Order to the Web." (1999)'
-        self.assertEqual(pretty.bib_oneliner(bibdata['Page99']), line)
+        self.assertEqual(color.undye(pretty.bib_oneliner(bibdata['Page99'])), line)
 
     def test_oneliner_no_year(self):
         decoder = endecoder.EnDecoder()
         bibdata = decoder.decode_bibdata(bibtex_raw0)
         bibdata['Page99'].pop('year')
         line = u'Page, Lawrence et al. "The PageRank Citation Ranking: Bringing Order to the Web."'
-        self.assertEqual(pretty.bib_oneliner(bibdata['Page99']), line)
+        self.assertEqual(color.undye(pretty.bib_oneliner(bibdata['Page99'])), line)
 
 if __name__ == '__main__':
     unittest.main()