small modification for weight