update to pass test