Remove mock because it did not work for that test.