Another option is to add the check here too.