Update wc_red.py

fixed small bug