diff --git a/tests/lint/trailing_whitespace.py b/tests/lint/trailing_whitespace.py
index 05d52e4c..f7d49e3c 100644
--- a/tests/lint/trailing_whitespace.py
+++ b/tests/lint/trailing_whitespace.py
@@ -1,14 +1,14 @@
 import itertools
 import re
 
-from tests.util import colon, color, display_path, paths, Failure, Success
+from tests.util import *
 
 
 def display_trailing_whitespace(whitespace):
     # Show trailing whitespace with printing characters and in red
     # To make it easy to see what needs to be removed
     replaced = whitespace.replace(' ', '-').replace('\t', r'\t')
-    return color(31, replaced)
+    return color(RED, replaced)
 
 
 def display_failure(failure):