step 17 - refactor test