fix labeling.py