Refactor test
Update stub data to reflect reality