adapt to the new weights files