Fixing weights bug pointed out by Chayanit