Clean up dependencies for mocking