diff --git a/scripts/generate_test_mask_image.py b/scripts/generate_test_mask_image.py
index 7af610a9bd20..313f560fbeef 100755
--- a/scripts/generate_test_mask_image.py
+++ b/scripts/generate_test_mask_image.py
@@ -8,7 +8,7 @@
 import os
 import sys
 import argparse
-from PyQt.QtGui import QImage, QColor
+from PyQt.QtGui import QImage, QColor, qRed, qBlue, qGreen, qAlpha, qRgb
 import struct
 import urllib2
 import glob