Update to fix the azure deploy issue