Fix issue on prod script