Fix distribution script