This is pretty unclear. Could you make use of `assert_called_with`? Or would make the test too brittle?