fix for multi part train