Correct mistake in weight check