fix script for rescheduler train