Move note on checking expectations to mock section