Change the test in order to use the mockups
Merge pull request #8 from asgoodasnu/WK-731

Change the test in order to use the mockups