Add test function for new check