Deployment code fix