Fixed train bug