prox: fixing error in weights