Fix deployment issue