Change the test in order to use the mockups