refactor the first test