Re-add the check