Remove forced mock for tests that dont need it