main.py: fix comparison to output proper colors