Corrected a bug that makes GPDB41 fail in train output tests