Change mock to double

The mock is deprecated