Fix trais script