Fix problem with edit weight