Correct mistake in weight check
Merge pull request #25 from In4matik/experimental

Correct mistake in weight check