Fixed a test broken by the change in the output_weight