Slight modification to mock setup in CalculatorTest