Refactor tests with SitePrism