Update deployment.py to pass deployment test