Revert "remove check"

This reverts commit bf9847456bcfee5cdb4378a426a92ab23671cf86.