fix bug in cute train