modify example to pass test