Revert "remove check. roll back"

This reverts commit bb3a337d3bd9be465702f59445126827c51b0698.