Fix deployment code