Applied fix for #40 to train pages