Quick fix for train edit tool