Fixed another error in loading the weights.