add back in weight check