Update engine.py (bug fixed)