Minor change to mechanism.py