Allow mock overriding