Updating the die to use mocks