Fixed pipeline to ensure feature weights