Fix issue when deploying