Add condition in setter