Refactor by removing 1 check.