Clean up spec file now it works in mock.