Minor change in mechanism.py